RTCDet-v2-N

config/model_config/rtcdet_v2_config.py

@@ -10,8 +10,8 @@ rtcdet_v2_cfg = {
         'bk_act': 'silu',
         'bk_norm': 'BN',
         'bk_depthwise': False,
-        'width': 1.0,
-        'depth': 1.0,
+        'width': 0.25,
+        'depth': 0.34,
         'stride': [8, 16, 32],  # P3, P4, P5
         'max_stride': 32,
         ## Neck: SPP

train_single_gpu.sh

@@ -4,13 +4,13 @@ python train.py \
         -d coco \
         --root /data/datasets/ \
         -m rtcdet_v2_l \
-        -bs 16 \
+        -bs 64 \
         -size 640 \
         --wp_epoch 3 \
         --max_epoch 300 \
         --eval_epoch 10 \
         --no_aug_epoch 15 \
-        --grad_accumulate 1 \
+        --grad_accumulate 2 \
         --ema \
         --fp16 \
         --multi_scale \