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和bash统计CPU利用率的方法
Jul 10 Python
Python中Django框架利用url来控制登录的方法
Jul 25 Python
python的格式化输出(format,%)实例详解
Jun 01 Python
Python3实现的爬虫爬取数据并存入mysql数据库操作示例
Jun 06 Python
python简单鼠标自动点击某区域的实例
Jun 25 Python
Python3环境安装Scrapy爬虫框架过程及常见错误
Jul 12 Python
python GUI图形化编程wxpython的使用
Jul 19 Python
python定义类self用法实例解析
Jan 22 Python
为什么相对PHP黑python的更少
Jun 21 Python
Python如何批量生成和调用变量
Nov 21 Python
python中使用.py配置文件的方法详解
Nov 23 Python
FP-growth算法发现频繁项集——发现频繁项集
Jun 24 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中显示格式化的用户输入
2006/10/09 PHP
php结合js实现点击超链接执行删除确认操作
2014/10/31 PHP
PHP实现获取文件后缀名的几种常用方法
2015/08/08 PHP
PHP实现针对日期,月数,天数,周数,小时,分,秒等的加减运算示例【基于strtotime】
2017/04/19 PHP
php微信公众号开发之音乐信息
2018/10/20 PHP
初学prototype,发个JS接受URL参数的代码
2006/09/25 Javascript
JQuery动态创建DOM、表单元素的实现代码
2011/08/09 Javascript
jQuery插件datepicker 日期连续选择
2015/06/12 Javascript
node网页分段渲染详解
2016/09/05 Javascript
nodejs redis 发布订阅机制封装实现方法及实例代码
2016/12/15 NodeJs
Vue键盘事件用法总结
2017/04/18 Javascript
jQuery+Ajax实现用户名重名实时检测
2017/06/01 jQuery
深入理解JavaScript的async/await
2018/08/05 Javascript
微信小程序与后台PHP交互的方法实例分析
2018/12/10 Javascript
JavaScript强制类型转换和隐式类型转换操作示例
2019/05/01 Javascript
vue中利用Promise封装jsonp并调取数据
2019/06/18 Javascript
[02:11]2014DOTA2 TI专访VG战队Fenrir:队伍气氛良好
2014/07/11 DOTA
Python生成密码库功能示例
2017/05/23 Python
Python使用progressbar模块实现的显示进度条功能
2018/05/31 Python
Python for循环与range函数的使用详解
2019/03/23 Python
python3实现二叉树的遍历与递归算法解析(小结)
2019/07/03 Python
如何利用python给图片添加半透明水印
2019/09/06 Python
python 的topk算法实例
2020/04/02 Python
python 实现ping测试延迟的两种方法
2020/12/10 Python
Python爬取网站图片并保存的实现示例
2021/02/26 Python
html5+svg学习指南之SVG基础知识
2014/12/17 HTML / CSS
美国体育用品商店:Rally House(NCAA、NFL、MLB、NBA、NHL和MLS)
2018/01/03 全球购物
美国排名第一的葡萄酒俱乐部:Firstleaf Wine Club
2020/01/02 全球购物
DataReader和DataSet的异同
2014/12/31 面试题
后勤工作职责
2013/12/22 职场文书
竞争上岗演讲稿
2014/01/05 职场文书
五十岁生日宴会答谢词
2014/01/15 职场文书
出纳工作岗位责任制
2014/02/02 职场文书
人事助理自荐信
2014/02/02 职场文书
《改造我们的学习》心得体会
2014/11/07 职场文书
python基础之//、/与%的区别详解
2022/06/10 Python