码迷,mamicode.com
首页 > 其他好文 > 详细

pytorch掉坑记录:model.eval的作用

时间:2020-07-22 20:10:13      阅读:73      评论:0      收藏:0      [点我收藏+]

标签:输入   记录   one   mode   model   否则   数据   set   ase   

 

训练完train_datasets之后,model要来测试样本了。在model(test_datasets)之前,需要加上model.eval(). 否则的话,有输入数据,即使不训练,它也会改变权值。这是model中含有batch normalization层所带来的的性质。
在做one classification的时候,训练集和测试集的样本分布是不一样的,尤其需要注意这一点。

pytorch掉坑记录:model.eval的作用

标签:输入   记录   one   mode   model   否则   数据   set   ase   

原文地址:https://www.cnblogs.com/guweixin/p/13362066.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!