Some pytorch demo scripts based on mnist datasets, so that everyone can get started with the pytorch deep learning framework.
- Python3
- PyTorch~=1.0.0
- TorchVision
- Scikit-Image
cd single_cpu
python main.py -b 256 -e 20
cd single_gpu
python main.py -b 256 -e 20 -g 1
cd data_parallel
python main.py -b 256 -e 20 -g 2\3\4
pass
- batch size: 256
- batch time: 6s
- training time: 5min
- gpu util: 99%
- gpu memory: 10G