概述
(1) model.train()
在使用pytorch构建神经网络的时候，训练过程中会在程序上方添加一句model.train()，作用是启用BatchNormalization和 Dropout，将BatchNormalization和Dropout置为True
(2) model.eval()
不启用 BatchNormalization 和 Dropout，将BatchNormalization和Dropout置为False
最后
以上就是高挑耳机为你收集整理的pytorch中的 model.train()和 model.eval()的全部内容，希望文章能够帮你解决pytorch中的 model.train()和 model.eval()所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
![[tensorflow] eval和run的区别](/uploads/reation/bcimg4.png)
发表评论 取消回复