Neural Networks with PyTorch – John Robin Inston

In this note we investigate how to implement neural networks in python using ‘PyTorch’ for recognizing handwritten digits using the MNIST dataset. We shall be following the guide MNIST Handwritten Digit Recognition in PyTorch provided by Gregor Koehler in his article on Nextjournal.

Background

# Packages
import torch
import torchvision