@@ -3,15 +3,17 @@ python train.py \
--cuda \
-d coco \
--root /mnt/share/ssd2/dataset/ \
- -m yolox_m \
+ -m rtdetr_n \
-bs 16 \
-size 640 \
--wp_epoch 3 \
- --max_epoch 300 \
+ --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