LeNet-5 Implementation Using Pytorch

This project implements LeNet-5 on MNIST dataset.

Dataset

MNIST: http://yann.lecun.com/exdb/mnist/

Testing Environment

• Pytorch version: 1.0.0
• CUDA version: 9.0.176
• Python version: 3.6.8

Reference

• 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