We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d196fa5 commit 56b9031Copy full SHA for 56b9031
distributed_train.sh
@@ -1,5 +1,5 @@
1
#!/bin/bash
2
NUM_PROC=$1
3
shift
4
-torchrun --nproc-per-node=$NUM_PROC train.py "$@"
+torchrun --nproc_per_node=$NUM_PROC train.py "$@"
5
0 commit comments