mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
6 lines
108 B
Bash
Executable File
6 lines
108 B
Bash
Executable File
#!/bin/bash
|
|
NUM_PROC=$1
|
|
shift
|
|
python3 -m torch.distributed.launch --nproc_per_node=$NUM_PROC train.py "$@"
|
|
|