Update config.py

config.py

@@ -31,8 +31,13 @@
 if mode == "train":
     # Dataset
     # Image format
-    train_image_dir = f"data/T91_General100/train"
-    valid_image_dir = f"data/T91_General100/valid"
+    train_image_dir = f"data/TG191/train"
+    valid_image_dir = f"data/TG191/valid"
+    # LMDB format
+    train_lr_lmdb_path = f"data/train_lmdb/TG191_LRbicx{upscale_factor}_lmdb"
+    train_hr_lmdb_path = f"data/train_lmdb/TG191_HR_lmdb"
+    valid_lr_lmdb_path = f"data/valid_lmdb/TG191_LRbicx{upscale_factor}_lmdb"
+    valid_hr_lmdb_path = f"data/valid_lmdb/TG191_HR_lmdb"
 
     image_size = 72
     batch_size = 16
@@ -43,20 +48,20 @@
     start_epoch = 0
     resume_weight = ""
 
-    # Total number of epochs
+    # SGD optimizer use: . Adam optimizer use: .
     epochs = 600
 
     # Model optimizer parameter (less training and low PSNR)
-    # model_optimizer_name = "sgd"
-    # model_lr = 1e-3
-    # model_momentum = 0.9
-    # model_weight_decay = 1e-7
-    # model_nesterov = False
+    model_optimizer_name = "sgd"
+    model_lr = 1e-3
+    model_momentum = 0.9
+    model_weight_decay = 1e-7
+    model_nesterov = False
 
     # Modify optimizer parameter (faster training and better PSNR)
-    model_optimizer_name = "adam"
-    model_lr = 1e-3
-    model_betas = (0.9, 0.999)
+    # model_optimizer_name = "adam"
+    # model_lr = 1e-3
+    # model_betas = (0.9, 0.999)
 
 # ==============================================================================
 # Verify configuration