https://dlcoder.medium.com/using-pytorch-in-python-an-introduction-to-machine-learning-and-deep-learning-fdebab79356c