We use a ResNet18 architecture as described in [1]}. Since there are only 10 classes in CIFAR-10, we replace the last fully-connected layer so that the output is a vector of size 10. The rest of the architecture is left unchanged.
