updated to torchrun (#116)

facebookresearch · Jun 14, 2023 · b3cd99c · b3cd99c
1 parent e582d2e
commit b3cd99c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/bevt/ootb/run_bevt_train.sh b/benchmarks/bevt/ootb/run_bevt_train.sh
@@ -49,7 +49,7 @@ if [ -f ${work_dir}/latest.pth ]; then rm ${work_dir}/*.pth; fi
 
 (
     set -x
-    python -m torch.distributed.launch --nnodes ${NODE_COUNT} \
+    torchrun --nnodes ${NODE_COUNT} \
         --node_rank ${RANK} \
         --master_addr ${MASTER_ADDR} \
         --master_port ${MASTER_PORT} \