https://medium.com/@ekbergasimov/an-introduction-to-deep-learning-with-pytorch-a-hands-on-guide-81ef7e1a5030
