RT-ODLab/train.sh

# Train YOLO
python train.py \
        --cuda \
        -d voc \
        --root /mnt/share/ssd2/dataset/ \
        -m rtdetr_n \
        -bs 16 \
        -size 640 \
        --wp_epoch 1 \
        --max_epoch 150 \
        --eval_epoch 10 \
        --ema \
        --fp16 \
        --multi_scale \
        --mosaic 0 \
        --mixup 0
        # --resume weights/coco/yolox_s/yolox_s_best.pth \
        # --pretrained weights/coco/yolo_free_medium/yolo_free_medium_39.46.pth \
        # --eval_first