pytorch-image-models/distributed_train.sh

mirror of https://github.com/huggingface/pytorch-image-models.git
#!/bin/bash
NUM_PROC=$1
shift
torchrun --nproc-per-node=$NUM_PROC train.py "$@"
Distributed (multi-process) train, multi-gpu single process train, and NVIDIA AMP support 2019-04-06 01:51:39 +08:00			`#!/bin/bash`
			`NUM_PROC=$1`
			`shift`
Switch to torchrun as old launcher changed arg names in 2.0.. Fix #1728 2023-03-20 23:15:40 +08:00			`torchrun --nproc-per-node=$NUM_PROC train.py "$@"`
Distributed (multi-process) train, multi-gpu single process train, and NVIDIA AMP support 2019-04-06 01:51:39 +08:00