update lr

Author: JavaZeroo

config.py

@@ -14,7 +14,7 @@ def __init__(self, ):
         self.DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
         self.BATCH_SIZE = self.get_batch_size()
         # print(f'Batch Size: {self.BATCH_SIZE}')
-        self.ONE_CYCLE_MAX_LR = 0.0001
+        self.ONE_CYCLE_MAX_LR = 0.01
         self.MODEL_PATH = Path('model')
         self.check_path(self.MODEL_PATH)
 
@@ -30,6 +30,16 @@ def get_batch_size(self, ):
         else:
             BATCH_SIZE = 2
         return BATCH_SIZE
+
+    def print_config(self):
+        print(f'SOURCE_DIR: {self.SOURCE_DIR}')
+        print(f'TARGET_DIR: {self.TARGET_DIR}')
+        print(f'BATCH_SIZE: {self.BATCH_SIZE}')
+        print(f'NUM_EPOCHS: {self.NUM_EPOCHS}')
+        print(f'N_FOLD: {self.N_FOLD}')
+        print(f'CROP_RATIO: {self.CROP_RATIO}')
+        print(f'DEVICE: {self.DEVICE}')
+        print(f'ONE_CYCLE_MAX_LR: {self.ONE_CYCLE_MAX_LR}')
 
     def check_path(self, path):
         path.mkdir(exist_ok=True)

train.py

@@ -24,6 +24,7 @@
 from utils import *
 
 config = Config()
+config.print_config()
 os.environ['WANDB_API_KEY'] = '67c99389e1ae37b747c40634c51802a4bf019d49'
 
 def train_transform():