yjh0410 1 year ago
parent
commit
822b71c138
1 changed files with 2 additions and 2 deletions
  1. 2 2
      yolo/config/fcos_config.py

+ 2 - 2
yolo/config/fcos_config.py

@@ -50,9 +50,9 @@ class FcosBaseConfig(object):
         # ---------------- Optimizer config ----------------
         self.trainer      = 'yolo'
         self.optimizer    = 'adamw'
-        self.base_lr      = 0.001     # base_lr = per_image_lr * batch_size
+        self.base_lr      = 0.0001     # base_lr = per_image_lr * batch_size
         self.min_lr_ratio = 0.01      # min_lr  = base_lr * min_lr_ratio
-        self.batch_size_base = 64
+        self.batch_size_base = 16
         self.momentum     = 0.9
         self.weight_decay = 0.05
         self.clip_max_norm   = 35.0