This is implementation of PyTorch imagenet classifier examples, with multi-GPU and multi-node supported.
- Run single-GPU training:
bash run.sh
- Run multi-GPU training:
bash run.sh multi
- Run distributed training:
bash run.sh dist
(changes in dist_config.txt required)