Pytorch实践--eval 模式,每次预测相同的数据获得的几率不一样

描述:使用pytorch训练好的模型对数据进行预测时,将model设置为eval()模式以后,发现一样的数据每次预测时的几率值会变更。web 分析:eval()能够将模型切换为 evaluation 模式,(仅仅当模型中含有Dropout和BatchNorm才起做用)。这里所说的 “模型中含有Dropout和BatchNorm ” 是指 包含在*_modules* 中(当类成员的type派生于Mo
相关文章
相关标签/搜索