mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
Change torchrun args to use _ instead of -, - is the new format, but looks like _ still works for backward compat with old versions. Fix #1742
This commit is contained in:
parent
d196fa536d
commit
56b90317cd
@ -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 "$@"
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user