Fix: Fix the reference parameter name error

Fix the reference parameter name error. The name is "SR_IMAGE_TYPE" instead of "IMAGE_TYPE"
chairc · Nov 1, 2024 · e1ba4c7 · e1ba4c7
1 parent 8901191
commit e1ba4c7
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 ******