DigitClassifier-PyTorch Simple Neural Network that classifies a digit by looking at an image of a hand-written digit