Peter_Featherstone:

What’s weird is the loss function of the classifier decreases when the model is training, i.e. when classifier.train() is called, but not when it is evaluated, i.e. when classifier.eval() is called.


I am pretty sure that this is expected behavior. .eval() changes the behavior of certain Module supertypes. Dropout, for instance, becomes the identity function when that method is called on it.