suryansh-sinha / LeNet5-MNIST Public

Notifications You must be signed in to change notification settings
Fork 0
Star 0

Implementation of LeNet-5 on MNIST Dataset in PyTorch

0 stars 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 5 Commits
images		images
.DS_Store		.DS_Store
LeNet.ipynb		LeNet.ipynb
README.md		README.md

Repository files navigation

LeNet5 PyTorch Implementation

This is a simple implementation of the paper "Gradient-Based Learning Applied to Document Recognition" by LeCun et. al in PyTorch. I have also tried to improve the model using ReLU instead of Tanh and max-pooling instead of average-pooling.

Feel free to ask anything!

Architecture

Requirments

Python3
PyTorch >= 2.0.0
torchvision >= 0.14.0

Accuracy

Average precision on test set: 98%

About

Implementation of LeNet-5 on MNIST Dataset in PyTorch

cnn pytorch mnist mnist-classification lenet convolutional-neural-networks convolutional-neural-network lenet-5 lenet-mnist pytorch-implementation

Report repository

Releases

No releases published

Packages

No packages published

Languages

Jupyter Notebook 100.0%