mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
6 lines
81 B
Bash
Executable File
6 lines
81 B
Bash
Executable File
#!/bin/bash
|
|
NUM_PROC=$1
|
|
shift
|
|
torchrun --nproc-per-node=$NUM_PROC train.py "$@"
|
|
|