Check dropout is deactive

vision
2

Yes, the model still contains the dropout layer, which is why it’ll be printed, but would disable it.
You can check its training attribute, which should be set to False, via: print(model.dropout_layer.training).