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

 更新时间:2020年06月23日 08:35:18   作者:youuuuuziiii  
这篇文章主要介绍了pytorch掉坑记录:model.eval的作用说明,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧

训练完train_datasets之后,model要来测试样本了。在model(test_datasets)之前,需要加上model.eval(). 否则的话,有输入数据,即使不训练,它也会改变权值。

这是model中含有batch normalization层所带来的的性质。

在做one classification的时候,训练集和测试集的样本分布是不一样的,尤其需要注意这一点。

补充知识:pytorch测试的时候为何要加上model.eval()

Do need to use model.eval() when I test?

Sure, Dropout works as a regularization for preventing overfitting during training.

It randomly zeros the elements of inputs in Dropout layer on forward call.

It should be disabled during testing since you may want to use full model (no element is masked)

使用PyTorch进行训练和测试时一定注意要把实例化的model指定train/eval,eval()时,框架会自动把BN和DropOut固定住,不会取平均,而是用训练好的值,不然的话,一旦test的batch_size过小,很容易就会被BN层导致生成图片颜色失真极大!!!!!!

以上这篇pytorch掉坑记录:model.eval的作用说明就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • Python3.5迭代器与生成器用法实例分析

    Python3.5迭代器与生成器用法实例分析

    这篇文章主要介绍了Python3.5迭代器与生成器用法,结合实例形式分析了Python3.5列表生成式、生成器、迭代器等相关原理与用法,需要的朋友可以参考下
    2019-04-04
  • Python基于回溯法子集树模板解决野人与传教士问题示例

    Python基于回溯法子集树模板解决野人与传教士问题示例

    这篇文章主要介绍了Python基于回溯法子集树模板解决野人与传教士问题,简单说明了野人与传教士问题,并结合实例形式分析了Python使用回溯法子集树模板解决野人与传教士问题的步骤与相关操作技巧,需要的朋友可以参考下
    2017-09-09
  • python目录与文件名操作例子

    python目录与文件名操作例子

    这篇文章主要介绍了python目录与文件名操作例子,需要的朋友可以参考下
    2016-08-08
  • Python实现输出程序执行进度百分比的方法

    Python实现输出程序执行进度百分比的方法

    这篇文章主要介绍了Python实现输出程序执行进度百分比的方法,涉及Python数值运算与系统输出相关操作技巧,需要的朋友可以参考下
    2017-09-09
  • Django解决无法从request.POST中获取URL传进来的参数

    Django解决无法从request.POST中获取URL传进来的参数

    这篇文章主要介绍了Django解决无法从request.POST中获取URL传进来的参数问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2023-12-12
  • 寻找网站后台地址的python脚本

    寻找网站后台地址的python脚本

    这篇文章主要介绍了用python实现的寻找网站后台地址的脚本代码,国外牛人的作品,需要的朋友可以参考下
    2014-09-09
  • Python调整matplotlib图片大小的3种方法汇总

    Python调整matplotlib图片大小的3种方法汇总

    我们在使用matplotlib作图时,会遇到图片不清晰或者图片大小不是我们想要的,这个时候就需要调整下,这篇文章主要给大家介绍了关于Python调整matplotlib图片大小的3种方法,文中通过实例代码介绍的非常详细,需要的朋友可以参考下
    2022-08-08
  • 深入学习python的yield和generator

    深入学习python的yield和generator

    这篇文章主要为大家详细介绍了python的yield和generator,针对python的生成器和yield关键字进行深入学习,感兴趣的小伙伴们可以参考一下
    2016-03-03
  • Pytest运行及其控制台输出信息

    Pytest运行及其控制台输出信息

    这篇文章主要介绍了Pytest运行及其控制台输出信息,文章围绕主题展开详细的内容介绍,具有一定的参考价值,需要的小伙伴可以参考一下
    2022-09-09
  • Python中six模块基础用法

    Python中six模块基础用法

    在本篇文章里小编给大家分享的是关于Python中six模块基础用法以及相关知识点,需要的朋友们学习下。
    2019-12-12

最新评论