pytorch:model.train和model.eval用法及区别详解


Posted in Python onFebruary 20, 2020

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

Class Inpaint_Network()
 ......

Model = Inpaint_Nerwoek()
 
#train:

Model.train(mode=True)

.....

#test:

Model.eval()

model.train()

启用 BatchNormalization 和 Dropout

model.eval()

不启用 BatchNormalization 和 Dropout

训练完train样本后,生成的模型model要用来测试样本。在model(test)之前,需要加上model.eval(),否则的话,有输入数据,即使不训练,它也会改变权值。这是model中含有batch normalization层所带来的的性质。

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

以上这篇pytorch:model.train和model.eval用法及区别详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
使用IPython来操作Docker容器的入门指引
Apr 08 Python
Python 功能和特点(新手必学)
Dec 30 Python
python中多个装饰器的执行顺序详解
Oct 08 Python
使用django-guardian实现django-admin的行级权限控制的方法
Oct 30 Python
利用python在excel里面直接使用sql函数的方法
Feb 08 Python
Django实现学生管理系统
Feb 26 Python
python程序快速缩进多行代码方法总结
Jun 23 Python
Python使用python-docx读写word文档
Aug 26 Python
Python 函数绘图及函数图像微分与积分
Nov 20 Python
Django2 连接MySQL及model测试实例分析
Dec 10 Python
python实现ping命令小程序
Dec 28 Python
Python Process创建进程的2种方法详解
Jan 25 Python
浅谈PyTorch的可重复性问题(如何使实验结果可复现)
Feb 20 #Python
pytorch 模型的train模式与eval模式实例
Feb 20 #Python
pytorch dataloader 取batch_size时候出现bug的解决方式
Feb 20 #Python
pytorch 使用加载训练好的模型做inference
Feb 20 #Python
pytorch中的inference使用实例
Feb 20 #Python
python encrypt 实现AES加密的实例详解
Feb 20 #Python
Python关于反射的实例代码分享
Feb 20 #Python
You might like
php+js实现图片的上传、裁剪、预览、提交示例
2013/08/27 PHP
typecho插件编写教程(四):插件挂载
2015/05/28 PHP
在WordPress中使用PHP脚本来判断访客来自什么国家
2015/12/10 PHP
php session实现多级目录存放实现代码
2016/02/03 PHP
PHP多线程模拟实现秒杀抢单
2018/02/07 PHP
PHP实现的大文件切割与合并功能示例
2018/04/10 PHP
laravel框架上传图片实现实时预览功能
2019/10/14 PHP
PHP使用PhpSpreadsheet操作Excel实例详解
2020/03/26 PHP
TP5框架页面跳转样式操作示例
2020/04/05 PHP
Extjs 几个方法的讨论
2010/01/28 Javascript
js日期范围初始化得到前一个月日期的方法
2015/05/05 Javascript
Jquery和angularjs获取check框选中的值的方法汇总
2016/01/17 Javascript
js创建数组的简单方法
2016/07/27 Javascript
js插件dropload上拉下滑加载数据实例解析
2016/07/27 Javascript
angular ngClick阻止冒泡使用默认行为的方法
2016/11/03 Javascript
Angular2学习笔记——详解路由器模型(Router)
2016/12/02 Javascript
JS实现的添加弹出层并完成锁屏操作示例
2017/04/07 Javascript
Angular4学习笔记router的简单使用
2018/03/30 Javascript
总结JavaScript在IE9之前版本中内存泄露问题
2018/04/28 Javascript
JS实现随机点名器
2020/04/12 Javascript
JS错误处理与调试操作实例分析
2020/04/13 Javascript
JavaScript实现页面高亮操作提示和蒙板
2021/01/04 Javascript
pyqt5 键盘监听按下enter 就登陆的实例
2019/06/25 Python
Python3.9新特性详解
2020/10/10 Python
兰蔻加拿大官方网站:Lancome加拿大
2016/08/05 全球购物
小型女装店的创业计划书
2014/01/09 职场文书
大学毕业自我鉴定范文
2014/02/03 职场文书
暑期培训随笔感言
2014/03/10 职场文书
师德建设实施方案
2014/03/21 职场文书
领导干部群众路线对照检查材料
2014/11/05 职场文书
街道党风廉政建设调研报告
2015/01/01 职场文书
英文慰问信范文
2015/03/24 职场文书
2015年机关后勤工作总结
2015/05/26 职场文书
小兵张嘎观后感300字
2015/06/03 职场文书
2016年春季运动会广播稿
2015/08/19 职场文书
为Centos安装指定版本的Docker
2022/04/01 Servers