Underwhelming performance in neural network with PyTorch

Thank you for the link.