https://capressocoffemakerthecheapest.blogspot.com/2022/06/pytorch-nn-module-generalization.html