https://fairytalefashionheadband.blogspot.com/2024/08/pytorch-nn-module-generalization.html