这篇文章主要介绍了Pytorch中关于model.eval()的作用及分析,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
model.eval()的作用及分析
model.eval()
作用等同于self.train(False)
简而言之,就是评估模式。而非训练模式。
在评估模式下,batchNorm层,dropout层等用于优化训练而添加的网络层会被关闭,从而使得评估时不会发生偏移。
结论
在对模型进行评估时,应该配合使用with torch.no_grad() 与 model.eval():
1 2 3 4 5 6 7 |
|