1. 首页
  2. 文章中心
  3. Other

pytorch中的 model.train()和 model.eval()

148 阅读 0 评论
我是靠谱客的博主 高挑耳机，这篇文章主要介绍pytorch中的 model.train()和 model.eval()，现在分享给大家，希望可以做个参考。

(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中的内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
点赞(98)

相关文章

python中eval和float_python中eval与json.loads对json的处理
python中eval和float_python中eval与json.loads对json的处理
[tensorflow] eval和run的区别
[tensorflow] eval和run的区别
Eval是什么Eval是什么
Eval是什么Eval是什么
python卷积神经网络应用
python卷积神经网络应用
pytorch中的 model.train()和 model.eval()
pytorch中的 model.train()和 model.eval()
TensorFlow：实战Google深度学习框架（三）深层神经网络第四章 深层神经网络
TensorFlow：实战Google深度学习框架（三）深层神经网络第四章 深层神经网络
eval
eval
Eval()的用法,有点不是很理解eval 方法
Eval()的用法,有点不是很理解eval 方法

评论列表共有 0 条评论

立即
投稿
返回
顶部