Bladeren bron

modify yolov7

yjh0410 1 jaar geleden
bovenliggende
commit
969e7c9990
2 gewijzigde bestanden met toevoegingen van 2 en 2 verwijderingen
  1. 1 1
      config/yolov7_config.py
  2. 1 1
      config/yolov8_config.py

+ 1 - 1
config/yolov7_config.py

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

+ 1 - 1
config/yolov8_config.py

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