Skip to content

Commit 7f756c3

Browse files
committed
add ddp script
1 parent 29f6cd6 commit 7f756c3

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

ddp_script.sh

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
export CUDA_VISIBLE_DEVICES=0,...,N_GPUS
2-
export NCCL_DEBUG=INFO
2+
# export NCCL_DEBUG=INFO
33
export NCCL_SOCKET_IFNAME=^docker0,lo
4-
export MASTER_ADDR="MASTER_IP"
5-
export MASTER_PORT="MASTER_PORT"
4+
export MASTER_ADDR=MASTER_IP
5+
export MASTER_PORT=MASTER_PORT
66

77
python3 src/main.py -t -e -rm_API -c CONFIG_PATH -DDP -n NUM_NODES -nr CURRENT_NODES -eval_type EVAL_TYPE ...

0 commit comments

Comments
 (0)