pytorch-image-models/distributed_train.sh