/pytorch-imageclassification-distributed

Primary LanguagePython

Pytorch image classification distributed

Pyotch distributed data parallel usage for image classification

python -m torch.distributed.launch --nproc_pre_node=4 train.py --datadir ''