https://wheretobuyiphone3gchargers.blogspot.com/2024/02/pytorch-nn-module-generalization.html