Merge pull request #96 from bestl1fe/dev

Fix: Fix the reference parameter name error
chairc · Nov 1, 2024 · 71754f8 · 71754f8
2 parents 8901191 + e1ba4c7
commit 71754f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/config/setting.py b/config/setting.py
@@ -24,7 +24,7 @@
 
 # model.networks.sr
 SR_IMAGE_TYPE = "RGB"
-SR_IMAGE_CHANNEL = image_type_choices[IMAGE_TYPE]
+SR_IMAGE_CHANNEL = image_type_choices[SR_IMAGE_TYPE]
 
 # Data processing
 # ****** torchvision.transforms.Compose ******