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