pytorch-image-models/distributed_train.sh

#!/bin/bash
NUM_PROC=$1
shift
python3 -m torch.distributed.launch --nproc_per_node=$NUM_PROC train.py "$@"
Distributed (multi-process) train, multi-gpu single process train, and NVIDIA AMP support 2019-04-05 10:51:39 -07:00			`#!/bin/bash`
			`NUM_PROC=$1`
			`shift`
Change the Python interpreter to Python 3.x in the scripts 2021-02-09 21:20:28 +02:00			`python3 -m torch.distributed.launch --nproc_per_node=$NUM_PROC train.py "$@"`
Distributed (multi-process) train, multi-gpu single process train, and NVIDIA AMP support 2019-04-05 10:51:39 -07:00