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 相关文章推荐
python的re模块应用实例
Sep 26 Python
Python使用scrapy采集时伪装成HTTP/1.1的方法
Apr 08 Python
Python使用shelve模块实现简单数据存储的方法
May 20 Python
python字典基本操作实例分析
Jul 11 Python
Python爬虫模拟登录带验证码网站
Jan 22 Python
Tensorflow环境搭建的方法步骤
Feb 07 Python
python读取excel指定列数据并写入到新的excel方法
Jul 10 Python
python绘制多个曲线的折线图
Mar 23 Python
Python数据可视化之画图
Jan 15 Python
用Python去除图像的黑色或白色背景实例
Dec 12 Python
django 文件上传功能的相关实例代码(简单易懂)
Jan 22 Python
用python修改excel表某一列内容的操作方法
Jun 11 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对包含html标签的字符串进行截取的函数分享
2014/06/19 PHP
php抽象类和接口知识点整理总结
2019/08/02 PHP
php根据地址获取百度地图经纬度的实例方法
2019/09/03 PHP
php设计模式之观察者模式定义与用法经典示例
2019/09/19 PHP
细品javascript 寻址,闭包,对象模型和相关问题
2009/04/27 Javascript
Jquery+ajax请求data显示在GridView上(asp.net)
2010/08/27 Javascript
设为首页和收藏的Javascript代码(亲测兼容IE,Firefox,chrome等浏览器)
2013/11/18 Javascript
jQuery中使用Ajax获取JSON格式数据示例代码
2013/11/26 Javascript
javascript 自定义回调函数示例代码
2014/09/26 Javascript
JavaScript数组常用方法
2015/03/02 Javascript
JavaScript获取当前日期是星期几的方法
2015/04/06 Javascript
JavaScript对象属性检查、增加、删除、访问操作实例
2015/07/08 Javascript
基于javascript实现文字无缝滚动效果
2016/03/22 Javascript
详解Vue2+Echarts实现多种图表数据可视化Dashboard(附源码)
2017/03/21 Javascript
JS简单实现数组去重的方法示例
2017/03/27 Javascript
如何选择jQuery版本 1.x? 2.x? 3.x?
2017/04/01 jQuery
vue模仿网易云音乐的单页面应用
2019/04/24 Javascript
JavaScript面向对象程序设计中对象的定义和继承详解
2019/07/29 Javascript
[01:04:05]Mineski vs TNC 2019国际邀请赛小组赛 BO2 第一场 8.15
2019/08/16 DOTA
python实现textrank关键词提取
2018/06/22 Python
Django框架设置cookies与获取cookies操作详解
2019/05/27 Python
GDAL 矢量属性数据修改方式(python)
2020/03/10 Python
Python爬虫之App爬虫视频下载的实现
2020/12/08 Python
移动端HTML5开发神器之vconsole详解
2020/12/15 HTML / CSS
韩国三大免税店之一:THE GRAND 中文免税店
2016/07/21 全球购物
世嘉游戏英国官方商店:SEGA Shop UK
2019/09/20 全球购物
法国包包和行李箱销售网站:Bagage24.fr
2020/03/24 全球购物
法学专业毕业生自荐信范文
2013/12/18 职场文书
家长对孩子的感言
2014/03/10 职场文书
关于运动会广播稿200字
2014/10/08 职场文书
党的群众路线教育实践活动总结大会主持词
2014/10/30 职场文书
离婚被告代理词
2015/05/23 职场文书
Java Socket实现Redis客户端的详细说明
2021/05/26 Redis
Python Django框架介绍之模板标签及模板的继承
2021/05/27 Python
Python OpenCV 图像平移的实现示例
2021/06/04 Python
基于Go语言构建RESTful API服务
2021/07/25 Golang