소스 검색

add train_ddp.sh file

yjh0410 2 년 전
부모
커밋
4e4666f514
1개의 변경된 파일15개의 추가작업 그리고 0개의 파일을 삭제
  1. 15 0
      train_ddp.sh

+ 15 - 0
train_ddp.sh

@@ -0,0 +1,15 @@
+# 8 GPUs
+python -m torch.distributed.run --nproc_per_node=8 train.py \
+                                                    --cuda \
+                                                    -dist \
+                                                    -d coco \
+                                                    --root /data/datasets/ \
+                                                    -m yolov1 \
+                                                    -bs 128 \
+                                                    -size 640 \
+                                                    --wp_epoch 1 \
+                                                    --max_epoch 150 \
+                                                    --eval_epoch 10 \
+                                                    --ema \
+                                                    --fp16 \
+                                                    --multi_scale \