yjh0410 2 年之前
父節點
當前提交
80aee3edd5
共有 1 個文件被更改,包括 1 次插入1 次删除
  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',
     },
 
 }