LeNet-scratch / README.md
ggsmith's picture
Update README.md
a017756 verified
metadata
license: mit
pipeline_tag: image-classification
tags:
  - LeNet
datasets:
  - uoft-cs/cifar10
language:
  - en
metrics:
  - accuracy

LeNet-5 From Scratch

This model was created using code referenced in Mastering Pytorch by Ashish Ranjan Jha. I modified the code to run on a GPU in Google Colab and I made a few more changes to the naming convention.

The notebook is available on GitHub.