The code in this repo formed part of a coursework completed for the 'Deep Learning' course. It is written using PyTorch, and is split into different parts.
A summary of the parts is as follows:
- Part 1: define the 2D convolution, max pooling operation, linear layer and 2D batch normalization
- Part 2: train a ResNet-18 on the CIFAR-10 dataset
- Part 3: answers to questions
The following ResNet architecture is built and trained to perform classification on the CIFAR-10 dataset
