Преглед на файлове

train fcos with 50 epoch

yjh0410 преди 10 месеца
родител
ревизия
d4d3917799
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      yolo/config/fcos_config.py

+ 1 - 1
yolo/config/fcos_config.py

@@ -63,7 +63,7 @@ class FcosBaseConfig(object):
         # ---------------- Lr Scheduler config ----------------
         self.warmup_iters = 500
         self.lr_scheduler = "cosine"
-        self.max_epoch    = 150
+        self.max_epoch    = 50
         self.eval_epoch   = 10
         self.no_aug_epoch = -1