瀏覽代碼

set yolov8's max epochs to 300

yjh0410 1 年之前
父節點
當前提交
81d248b7c2
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      yolo/config/yolov8_config.py

+ 1 - 1
yolo/config/yolov8_config.py

@@ -88,7 +88,7 @@ class Yolov8BaseConfig(object):
         # ---------------- Lr Scheduler config ----------------
         # ---------------- Lr Scheduler config ----------------
         self.warmup_epoch = 3
         self.warmup_epoch = 3
         self.lr_scheduler = "cosine"
         self.lr_scheduler = "cosine"
-        self.max_epoch    = 500
+        self.max_epoch    = 300
         self.eval_epoch   = 10
         self.eval_epoch   = 10
         self.no_aug_epoch = 20
         self.no_aug_epoch = 20