No description, website, or topics provided.
Python
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
README.md
main.py
requirements.txt

README.md

Basic MNIST Example

This code is a copy from:

https://github.com/pytorch/examples/tree/master/mnist

with slight modifications that fix problems with downloading the data.

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