|
@@ -3,7 +3,7 @@ python train.py \
|
|
|
--cuda \
|
|
--cuda \
|
|
|
-d coco \
|
|
-d coco \
|
|
|
--root /mnt/share/ssd2/dataset/ \
|
|
--root /mnt/share/ssd2/dataset/ \
|
|
|
- -m yolox_n \
|
|
|
|
|
|
|
+ -m yolox_s \
|
|
|
-bs 16 \
|
|
-bs 16 \
|
|
|
-size 640 \
|
|
-size 640 \
|
|
|
--wp_epoch 3 \
|
|
--wp_epoch 3 \
|
|
@@ -12,7 +12,7 @@ python train.py \
|
|
|
--ema \
|
|
--ema \
|
|
|
--fp16 \
|
|
--fp16 \
|
|
|
--multi_scale \
|
|
--multi_scale \
|
|
|
- --resume weights/coco/yolox_n/yolox_n_best.pth \
|
|
|
|
|
|
|
+ # --resume weights/coco/yolox_n/yolox_n_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