For demonstration we train it on the very common MNIST dataset of handwritten digits. In this code we go through how to create the network as well as initialize a loss function, optimizer, check accuracy and more.
To get started Make sure Python3 is installed on your machine. Set up a PyTorch virtual enviroment and edit your code as below