https://keroseneheaterreview.blogspot.com/2024/01/pytorch-how-to-deactivate-dropout-in.html