pytorch:model.train和model.eval用法及区别详解
作者：Hinton-wu 发布时间：2022-07-13 19:22:51
标签：pytorch,model.train,model.eval
使用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的时候，训练集和测试集的样本分布是不一样的，尤其需要注意这一点。
来源：https://blog.csdn.net/qq_38410428/article/details/101102075
0
投稿猜你喜欢
- 如果不清楚字符串的编码格式的话，就可以将这段字符这样检查：$encode = mb_detect_encoding($string, arr
- 1、灵活运用样式熟悉网页设计的网友就知道，调用Style的方法很多，我们可以单击鼠标右键选择Custo
- Python 3.7中一个令人兴奋的新特性是 data classes 。 数据类通常是一个主要包含数据的类，尽管实际上没有任何限制。 它是
- 第一种 使用pygame模块 pygame.mixer.init() pygame.mixer.music.load
- 1.相关基础python3中有两种字符串类型：str和bytespython编码问题可以参考文章str以unicode编码格式保存在内存所以
- 阅读上一篇：AJAX的jQuery实现入门(一)要写入数据库,我们知道的最简单的就是注册了, 就做个最简单的注册表单, 看看是如何提交数据的
- 这篇文章主要介绍了简单了解Python读取大文件代码实例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的
起因在公司搭建了套webpack多页面应用脚手架，开始用着很爽，解决了既想使用Vue的模块化开发，又想做多页打包上线管理的初衷，但是随着业务- 这些包可以独立使用，也可以与其他包一起使用以满足复杂的业务需求。Integration Services 可以提取和转换来自多种源（如 XM
- iUI、jQTouch、WPTouch、PhoneGap、XUI、iWebkit、Rhodes、gwt-mobile…当我们已经开始惊叹 w
- 可怜我的C盘本来只有8.XG，所以不得不卸载掉它。卸载掉本身没啥问题，只是昨晚突然发现 Sql Server 2008 R2 Managem
- PHP children() 函数实例查找 note 节点的子节点：<?php $note=<<<XML<no
- 首先我们知道这个效果应该是一个老话题了。今天整理文件的时候，发现自己以前的一些布局的解决方法躺在文件夹里很长时间了，翻翻老底吧。需要说明的是
- 前言python 可以做网站应用，也可以做客户端应用。但是客户端应用需要运行 py 脚本，如果用户不懂 python 就是一件比较麻烦的事情
- 内容摘要：Microsoft建立了一种既灵活又强大的安全管理机制，它能够对用户访问SQL Server服务器系统和数据库的安全进行
- 1 , javascript字符集：javascript采用的是Unicode字符集编码。为什么要采用这个编码呢？原因很简单，16位的Uni
- 使用 Microsoft® SQL Server™ 2000，可以选择在一台计算机上安装 SQL Ser
- 注：本文所说的视觉设计师专指网页视觉设计师。网页设计师与平面设计师都归类为设计师，其实这两个职业是跨行业的，虽然有很多设计师一直在跨行业工作
- 一、使用loadVariables 一个例子简单的描述了如何通过GET方法向服务器端的ASP发送请求: _root. pushAc
- 制作爬虫的步骤制作一个爬虫一般分以下几个步骤：分析需求分析网页源代码，配合开发者工具编写正则表达式或者XPath表达式正式编写 python
