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的作用说明就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • 使用C语言来扩展Python程序和Zope服务器的教程

    使用C语言来扩展Python程序和Zope服务器的教程

    这篇文章主要介绍了使用C语言来扩展Python程序和Zope服务器的教程,本文来自于IBM官方网站技术文档,需要的朋友可以参考下
    2015-04-04
  • Python中self关键字的用法解析

    Python中self关键字的用法解析

    在Python中,self是一个经常出现的关键字,特别是在类定义中的方法,这篇文章主要和大家self的作用和用法,希望可以帮助大家更好地理解为什么需要它以及如何正确使用它
    2023-11-11
  • 利用python3筛选excel中特定的行(行值满足某个条件/行值属于某个集合)

    利用python3筛选excel中特定的行(行值满足某个条件/行值属于某个集合)

    这篇文章主要给大家介绍了关于利用python3筛选excel中特定的行(行值满足某个条件/行值属于某个集合)的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2020-09-09
  • Python pyecharts数据可视化实例详解

    Python pyecharts数据可视化实例详解

    PyEcharts是一个用于生成 Echarts图表的类库, Python是一门富有表达力的语言,很适合用于数据处理,下面这篇文章主要给大家介绍了关于Python pyecharts数据可视化的相关资料,需要的朋友可以参考下
    2022-05-05
  • 如何理解python中数字列表

    如何理解python中数字列表

    在本篇文章里小编给大家分享的是关于python中数字列表知识点详解,有兴趣的朋友们可以参考下。
    2020-05-05
  • 七个Python必备的GUI库

    七个Python必备的GUI库

    这篇文章主要介绍了七个Python必备的GUI库,帮助大家更好的理解和学习使用python制作gui程序,感兴趣的朋友可以了解下
    2021-04-04
  • Django正则URL匹配实现流程解析

    Django正则URL匹配实现流程解析

    这篇文章主要介绍了Django正则URL匹配实现流程解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-11-11
  • Python检测PE所启用保护方式详解

    Python检测PE所启用保护方式详解

    Python通过pywin32模块调用WindowsAPI接口,可以实现对特定进程加载模块的枚举输出并检测该PE程序模块所启用的保护方式,感兴趣的可以了解一下
    2022-10-10
  • Python scrapy爬取起点中文网小说榜单

    Python scrapy爬取起点中文网小说榜单

    爬虫的基础内容已经全部学玩,博主决定想着更加标准化以及实用能力更强的scrapy进发,今天记录自己第一个scrapy爬虫项目. scrapy爬取起点中文网24小时热销榜单,需要的朋友可以参考下
    2021-06-06
  • 数据清洗--DataFrame中的空值处理方法

    数据清洗--DataFrame中的空值处理方法

    今天小编就为大家分享一篇数据清洗--DataFrame中的空值处理方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-07-07

最新评论