https://server.vpnwp.com/hyper-v/hyperscience-learn-first-steps-in-ml-with-pytorch-training-a-model/