Skip to content

Latest commit

 

History

History

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 

README.md

LeNet Architecture

LeNet.py contains the LeNet architecture.
mnist_main.py is a script that uses the above said architecture for training a digit classification model.

Notes

Do note that the original paper employed "Tanh" as the activation function. I have used "ReLU" here because of it's simplicity and improved performance.

References

Gradient-Based Learning Applied to Document Recognition