Skip to content

Commit 56b9031

Browse files
committed
Change torchrun args to use _ instead of -, - is the new format, but looks like _ still works for backward compat with old versions. Fix huggingface#1742
1 parent d196fa5 commit 56b9031

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

distributed_train.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
#!/bin/bash
22
NUM_PROC=$1
33
shift
4-
torchrun --nproc-per-node=$NUM_PROC train.py "$@"
4+
torchrun --nproc_per_node=$NUM_PROC train.py "$@"
55

0 commit comments

Comments
 (0)