pytorch掉坑记录：model.eval的作用
Posted guweixin
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了pytorch掉坑记录：model.eval的作用相关的知识，希望对你有一定的参考价值。
训练完train_datasets之后，model要来测试样本了。在model(test_datasets)之前，需要加上model.eval(). 否则的话，有输入数据，即使不训练，它也会改变权值。这是model中含有batch normalization层所带来的的性质。
在做one classification的时候，训练集和测试集的样本分布是不一样的，尤其需要注意这一点。
以上是关于pytorch掉坑记录：model.eval的作用的主要内容，如果未能解决你的问题，请参考以下文章
model.train()model.eval()optimizer.zero_grad()loss.backward()optimizer.step作用及原理详解Pytorch入门手册
Pytorch中的 model.train() 和 model.eval() 模式
pytorch踩坑之model.eval()和model.train()输出差距很大
使用 pytorch 训练和测试 CNN。有和没有 model.eval()