LeNet5-MNIST-PyTorch
LeNet5-MNIST-PyTorch copied to clipboard
ChawDoe
The simplest implementation of LeNet5 with mnist in PyTorch. Accuracy: ~99%
LeNet5-MNIST-PyTorch
This is the simplest implementation of the paper "Gradient-based learning applied to document recognition" in PyTorch.
Have a try with artificial intelligence!
Feel free to ask anything!
Requirments
Python3
PyTorch >= 0.4.0
torchvision >= 0.1.8
Usage
$git clone https://github.com/ChawDoe/LeNet-5-MNIST-PyTorch.git
$cd LeNet5-MNIST-PyTorch
$python3 train.py
Hint
This repo includes the mnist dataset.
Accuracy
Average precision on test set: 99%
ChawDoe