Browse Source

modify YOLOv8's config

yjh0410 2 years ago
parent
commit
1735d40e7a
1 changed files with 1 additions and 1 deletions
  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',
     },
 
 }