yjh0410 1 년 전
부모
커밋
408c0799a5
1개의 변경된 파일2개의 추가작업 그리고 1개의 파일을 삭제
  1. 2 1
      yolo/config/fcos_config.py

+ 2 - 1
yolo/config/fcos_config.py

@@ -51,7 +51,8 @@ class FcosBaseConfig(object):
         self.trainer      = 'simple'
         self.optimizer    = 'sgd'
         self.base_lr      = 0.01     # base_lr = per_image_lr * batch_size
-        self.min_lr_ratio = 0.01      # min_lr  = base_lr * min_lr_ratio
+        self.min_lr_ratio = 0.01     # min_lr  = base_lr * min_lr_ratio
+        self.bk_lr_ratio  = 1.0
         self.batch_size_base = 16
         self.momentum     = 0.9
         self.weight_decay = 0.0001