@@ -3,7 +3,7 @@ python train.py \
--cuda \
-d coco \
--root /mnt/share/ssd2/dataset/ \
- -m yolov7_large \
+ -m yolov5 \
-bs 16 \
-size 640 \
--wp_epoch 1 \