Basic MNIST Example

mnist/README.md

latest151 B

bash

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