https://videosashka.com/watch/vVrASMLVBnY/training-neural-networks-in-pytorch