diff --git a/distributed_train.sh b/distributed_train.sh
index 50938e91..fdd65d00 100755
--- a/distributed_train.sh
+++ b/distributed_train.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
 NUM_PROC=$1
 shift
-torchrun --nproc-per-node=$NUM_PROC train.py "$@"
+torchrun --nproc_per_node=$NUM_PROC train.py "$@"