https://srkswaohrktjhgto.blogspot.com/2021/06/nnmodels-pytorch-nn-package-pytorch.html