Apppublic

datacomp/ImageNetTraining0.0-frac-1over2

sourceHugging Faceupdated 2y agoView on Hugging Face

0likes

train.sh5 linesDownload Raw Back to root

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