Fix distributed train script

5 years ago · b0158a593e
parent 183d8e4aef
commit b0158a593e
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
-python -m torch.distributed.launch --nproc_per_node=$NUM_PROC dtrain.py "$@"
+python -m torch.distributed.launch --nproc_per_node=$NUM_PROC train.py "$@"