@@ -1,11 +1,11 @@
-# train YOLO with 4 GPUs
+# train YOLO with 8 GPUs
# 使用 8 GPU来训练YOLO
python -m torch.distributed.run --nproc_per_node=8 train.py \
--cuda \
-dist \
-d coco \
--root /data/datasets/ \
- -m yolovx_s \
+ -m yolovx_t \
-bs 128 \
-size 640 \
--wp_epoch 3 \