RT-ODLab/train_ddp.sh

# 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 yolov5_l \
                                                    -bs 128 \
                                                    -size 640 \
                                                    --wp_epoch 1 \
                                                    --max_epoch 300 \
                                                    --eval_epoch 10 \
                                                    --ema \
                                                    --fp16 \
                                                    --multi_scale \