首页 > 其他分享 >eval() train() lightning

eval() train() lightning

时间:2023-05-11 21:12:11浏览次数:36  
标签:end lightning train eval using model

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

相关文章