READ ME model to train a neural network with one hidden layer to classify images of hand-written digits using pytorch