Hi, I replaced the dropout layers by the funtion of the nn.functional API. The print of the state dict shows dropout.training. This behaviour is not as expected. I double checked my code to be sure that there is no network =  network.eval() at the wrong place.