This project implements LeNet-5 on MNIST dataset.
MNIST
: http://yann.lecun.com/exdb/mnist/
- Pytorch version:
1.0.0
- CUDA version:
9.0.176
- Python version:
3.6.8
MNISTtools.py
: Adapted by Charles Deledalle from https://gist.github.com/xmfbit. Inspired from http://abel.ee.ucla.edu/cvxopt/_downloads/mnist.py which is GPL licensed.
Note: If you have trouble reading the jupyter notebook in github, please click here