yjh0410 1 рік тому
батько
коміт
822b71c138
1 змінених файлів з 2 додано та 2 видалено
  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