This is a notebook for getting to learn Pytorch by building a play CNN and using it on the MNIST dataset
- Load dataset and labels
- Either use a standard dataset or write a custom dataset class
- Define dataloaders
- Build model and specify how you can build the forward pass
- Define training and evaluation loops