convnet implementation of a convolutional neural network using pytorch to perform classification on mnist Sample digits: Algorithm loss curve during training and testing (2 epochs):