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入门者的一些编程建议
Jun 15 Python
Python实现快速多线程ping的方法
Jul 15 Python
MySQL适配器PyMySQL详解
Sep 20 Python
Python中turtle作图示例
Nov 15 Python
python pands实现execl转csv 并修改csv指定列的方法
Dec 12 Python
Python实现的逻辑回归算法示例【附测试csv文件下载】
Dec 28 Python
python利用多种方式来统计词频(单词个数)
May 27 Python
python用线性回归预测股票价格的实现代码
Sep 04 Python
Python 矩阵转置的几种方法小结
Dec 02 Python
python3连接MySQL8.0的两种方式
Feb 17 Python
python 爬虫如何实现百度翻译
Nov 16 Python
基于PyTorch中view的用法说明
Mar 03 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
缅甸的咖啡简史
2021/03/04 咖啡文化
给WordPress中的留言加上楼层号的PHP代码实例
2015/12/14 PHP
Laravel 5.3 学习笔记之 安装
2016/08/28 PHP
PHPMailer发送邮件
2016/12/28 PHP
阻止JavaScript事件冒泡传递(cancelBubble 、stopPropagation)
2007/05/08 Javascript
给文字加上着重号的JS代码
2013/11/12 Javascript
jquery如何扑捉回车键触发的事件
2014/04/24 Javascript
JavaScript获取浏览器信息的方法
2015/11/20 Javascript
10个JavaScript中易犯小错误
2016/02/14 Javascript
js实现纯前端的图片预览
2016/04/27 Javascript
JS 通过系统时间限定动态添加 select option的实例代码
2016/06/09 Javascript
jQuery中text() val()和html()的区别实例详解
2016/06/28 Javascript
使用jQuery加载html页面到指定的div实现方法
2016/07/13 Javascript
vue.js通过自定义指令实现数据拉取更新的实现方法
2016/10/18 Javascript
JS实现鼠标移上去显示图片或微信二维码
2016/12/14 Javascript
详解能在多种前端框架下使用的表格控件
2017/01/11 Javascript
使用Vue.js和Element-UI做一个简单登录页面的实例
2018/02/23 Javascript
小程序云开发初探(小结)
2018/10/24 Javascript
vue跳转同一个组件,参数不同,页面接收值只接收一次的解决方法
2019/11/05 Javascript
基于javascript实现贪吃蛇经典小游戏
2020/04/10 Javascript
numpy.random模块用法总结
2019/05/27 Python
使用python快速实现不同机器间文件夹共享方式
2019/12/22 Python
python:HDF和CSV存储优劣对比分析
2020/06/08 Python
Nike台湾官方商店:Nike.com (TW)
2017/08/16 全球购物
以工厂直接定价的传奇性能:Ben Hogan Golf
2019/01/04 全球购物
Zavvi西班牙:电子游戏、极客服装、Blu-ray、Funko Pop等
2019/05/03 全球购物
Strathberry苏贝瑞中国官网:西班牙高级工匠手工打造
2020/10/19 全球购物
护士自我鉴定
2013/10/23 职场文书
环境保护与污染治理求职信
2014/07/16 职场文书
青春励志演讲稿范文
2014/08/25 职场文书
门卫岗位职责
2015/02/09 职场文书
go语言求任意类型切片的长度操作
2021/04/26 Golang
Python 流媒体播放器的实现(基于VLC)
2021/04/28 Python
微软PC Health Check电脑健康状况检查应用下载(Win11配置检测工具)
2021/06/26 数码科技
优化Mysql查询的示例
2022/04/26 MySQL
MySQL一劳永逸永久支持输入中文的方法实例
2022/08/05 MySQL