From e1ba4c7704314c032967d459fc65d613c135ce59 Mon Sep 17 00:00:00 2001
From: bestl1fe <166411263+bestl1fe@users.noreply.github.com>
Date: Fri, 1 Nov 2024 10:48:16 +0800
Subject: [PATCH] Fix: Fix the reference parameter name error

Fix the reference parameter name error. The name is "SR_IMAGE_TYPE" instead of "IMAGE_TYPE"
---
 config/setting.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/config/setting.py b/config/setting.py
index 6db5d4f..e905304 100644
--- 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 ******