训练完train_datasets之后,model要来测试样本了。在model(test_datasets)之前,需要加上model.eval(). 否则的话,有输入数据,即使不训练,它也会改变权值。(这样就导致测试结果不对)
这是model中含有batch normalization层所带来的的性质。
Pytorch笔记:torch.eval使用方法
最新推荐文章于 2024-08-18 12:24:05 发布
训练完train_datasets之后,model要来测试样本了。在model(test_datasets)之前,需要加上model.eval(). 否则的话,有输入数据,即使不训练,它也会改变权值。(这样就导致测试结果不对)
这是model中含有batch normalization层所带来的的性质。