use RTMDet Trainer for v2

config/model_config/rtcdet_v2_config.py

@@ -58,7 +58,7 @@ rtcdet_v2_cfg = {
         'loss_box_weight': {'tal': 7.0, 'ota': 5.0},
         'loss_dfl_weight': {'tal': 1.5, 'ota': 1.0},
         # ---------------- Train config ----------------
-        'trainer_type': 'yolox',
+        'trainer_type': 'rtmdet',
     },
 
 }

train_single_gpu.sh

@@ -2,9 +2,9 @@
 python train.py \
         --cuda \
         -d coco \
-        --root /mnt/share/ssd2/dataset/ \
-        -m yolox_s \
-        -bs 8 \
+        --root /data/datasets/ \
+        -m rtcdet_v2_l \
+        -bs 16 \
         -size 640 \
         --wp_epoch 3 \
         --max_epoch 300 \