mnist

and

Jan 16, 2019

0d3fe14 · Jan 16, 2019

This branch is 405 commits behind pytorch/examples:main.

Name	Name	Last commit message	Last commit date
parent directory ..
README.md	README.md	Add instruction to use specified GPU id (pytorch#73 )	Feb 21, 2017
main.py	main.py	Use argmax() instead of max()[1] in mnist/main.py (pytorch#494 )	Jan 16, 2019
requirements.txt	requirements.txt	Improve MNIST example	Jan 17, 2017

README.md

pip install -r requirements.txt
python main.py
# CUDA_VISIBLE_DEVICES=2 python main.py  # to specify GPU id to ex. 2