Switch to torchrun as old launcher changed arg names in 2.0.. Fix #1728

2023-03-20 08:15:40 -07:00 · 2023-03-20 08:15:40 -07:00 · 2054f11c6f
parent 49b9c3be80
commit 2054f11c6f
1 changed files with 1 additions and 1 deletions
--- a/distributed_train.sh
+++ b/distributed_train.sh
@ -1,5 +1,5 @@
 #!/bin/bash
 NUM_PROC=$1
 shift
-python3 -m torch.distributed.launch --nproc_per_node=$NUM_PROC train.py "$@"
+torchrun --nproc-per-node=$NUM_PROC train.py "$@"