https://discuss.pytorch.org/t/why-not-use-model-eval-in-training-step-method-on-lightning/122425/5
Issue:
I guess PL authors took care of switching between eval/train mode within pre defined callbacks… But problem is that when I try to predict test data in “on_fit_end” callback without using model.eval() it gives me different result than predicting outside training routine (and of course using model.eval in advance), thats why I’m wonder If ‘on_fit_end’ callback provided with model.eval … but I guess I chose inappropriate forum
标签:end,lightning,train,eval,using,model From: https://www.cnblogs.com/morganh/p/17392243.html