https://camera-optical-zoom.blogspot.com/2024/01/pytorch-nn-module-generalization.html