Thank you very much for your answer, you are correct!
It was a small mistake of mine, concerning the use of the same nn.BatchNorm2D, before and after nn.Conv2d in the model’s forward pass… Apparently, there was a conflict I didn’t know about, my bad!
Specifying each nn.BatchNorm2D solved the problem. Thanks!