Training a neural network to recognize handwritten digits (MNIST), using PyTorch. This repository includes all source code for the tutorial on DigitalOcean with the same title, including:
- MNIST training and evaluation code using a simple, custom neural network
- MNIST training and evaluation code using torchvision's ResNet18 architecture
- source code for associated Google Colab notebooks: Custom Arch notebook + ResNet18 notebook
created by Alvin Wan, December 2020