https://freehipwee.blogspot.com/2020/12/pytorch-deep-learning-and-artificial.html