yjh0410 2 years ago
parent
commit
80aee3edd5
1 changed files with 1 additions and 1 deletions
  1. 1 1
      config/model_config/rtcdet_v2_config.py

+ 1 - 1
config/model_config/rtcdet_v2_config.py

@@ -117,7 +117,7 @@ rtcdet_v2_cfg = {
         'loss_box_weight': 7.0,
         'loss_dfl_weight': 1.5,
         # ---------------- Train config ----------------
-        'trainer_type': 'rtmdet',
+        'trainer_type': 'yolov8',
     },
 
 }