Parcourir la source

modify YOLOv8's config

yjh0410 il y a 2 ans
Parent
commit
1735d40e7a
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      config/model_config/yolov8_config.py

+ 1 - 1
config/model_config/yolov8_config.py

@@ -99,7 +99,7 @@ yolov8_cfg = {
         'loss_box_weight': 7.5,
         'loss_dfl_weight': 1.5,
         # ---------------- Train config ----------------
-        'trainer_type': 'yolov8',
+        'trainer_type': 'rtcdet',
     },
 
 }