Added GPU support and stops training when test accuracy converges

ChawDoe / LeNet5-MNIST-PyTorch

The simplest implementation of LeNet5 with mnist in PyTorch. Accuracy: ~99%

MIT License

193 stars 56 forks source link

Closed samanthadurdy closed 1 year ago

samanthadurdy commented 1 year ago

Hi, I know this repo is pretty old but I added GPU support if you're interested

ChawDoe commented 1 year ago

Hi, thanks for your contribution. :)