Multiply weights after using dropout in training - PyTorch
PyTorch handles this with scaling the output of the dropout layer at training time with this probability:



So it handles this itself after applying model.eval(). You can lookup their code here: