|
@@ -3,15 +3,17 @@ python train.py \
|
|
|
--cuda \
|
|
--cuda \
|
|
|
-d coco \
|
|
-d coco \
|
|
|
--root /mnt/share/ssd2/dataset/ \
|
|
--root /mnt/share/ssd2/dataset/ \
|
|
|
- -m yolox_m \
|
|
|
|
|
|
|
+ -m rtdetr_n \
|
|
|
-bs 16 \
|
|
-bs 16 \
|
|
|
-size 640 \
|
|
-size 640 \
|
|
|
--wp_epoch 3 \
|
|
--wp_epoch 3 \
|
|
|
- --max_epoch 300 \
|
|
|
|
|
|
|
+ --max_epoch 150 \
|
|
|
--eval_epoch 10 \
|
|
--eval_epoch 10 \
|
|
|
--ema \
|
|
--ema \
|
|
|
--fp16 \
|
|
--fp16 \
|
|
|
--multi_scale \
|
|
--multi_scale \
|
|
|
|
|
+ --mosaic 0 \
|
|
|
|
|
+ --mixup 0
|
|
|
# --resume weights/coco/yolox_s/yolox_s_best.pth \
|
|
# --resume weights/coco/yolox_s/yolox_s_best.pth \
|
|
|
# --pretrained weights/coco/yolo_free_medium/yolo_free_medium_39.46.pth \
|
|
# --pretrained weights/coco/yolo_free_medium/yolo_free_medium_39.46.pth \
|
|
|
# --eval_first
|
|
# --eval_first
|
yjh0410