Spaces:

datacomp
/

ImageNetTraining80.0-frac-1over8

Paused

ImageNetTraining80.0-frac-1over8 / train.sh

meg HF Staff

Upload train.sh with huggingface_hub

b3e0a2b verified almost 2 years ago

207 Bytes

#!/bin/bash
NUM_PROC=$1
shift
# the localhost fixes are from https://github.com/pytorch/pytorch/issues/73320
torchrun  --rdzv_backend c10d --rdzv_endpoint localhost:0 --nproc_per_node=$NUM_PROC train.py "$@"

	#!/bin/bash
	NUM_PROC=$1
	shift
	# the localhost fixes are from https://github.com/pytorch/pytorch/issues/73320
	torchrun --rdzv_backend c10d --rdzv_endpoint localhost:0 --nproc_per_node=$NUM_PROC train.py "$@"