脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → pytorch:model.train和model.eval

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

 更新时间：2020年02月20日 17:31:39   作者：Hinton-wu  
今天小编就为大家分享一篇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的时候，训练集和测试集的样本分布是不一样的，尤其需要注意这一点。

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

您可能感兴趣的文章:

相关文章

  • 基于python爬虫数据处理(详解)

    基于python爬虫数据处理(详解)

    下面小编就为大家带来一篇基于python爬虫数据处理(详解)。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧
    2017-06-06
  • Python中for循环语句实战案例

    Python中for循环语句实战案例

    这篇文章主要给大家介绍了关于Python中for循环语句的相关资料,python中for循环一般用来迭代字符串,列表,元组等,当for循环用于迭代时不需要考虑循环次数,循环次数由后面的对象长度来决定,需要的朋友可以参考下
    2023-09-09
  • Python模块zipfile原理及使用方法详解

    Python模块zipfile原理及使用方法详解

    这篇文章主要介绍了Python模块zipfile原理及使用方法详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-08-08
  • python中的yield使用方法

    python中的yield使用方法

    这篇文章主要介绍了python中的yield使用方法,需要的朋友可以参考下
    2014-02-02
  • python装饰器三种装饰模式的简单分析

    python装饰器三种装饰模式的简单分析

    这篇文章主要介绍了python装饰器的三种装饰模式，帮助大家更好的理解和学习python 装饰器，感兴趣的朋友可以了解下
    2020-09-09
  • Django imgareaselect手动剪切头像实现方法

    Django imgareaselect手动剪切头像实现方法

    这篇文章主要介绍了Django imgareaselect手动剪切头像实现方法,实例分析了Django框架操作图片的相关技巧,需要的朋友可以参考下
    2015-05-05
  • 教你使用Pandas直接核算Excel中的快递费用

    教你使用Pandas直接核算Excel中的快递费用

    文中仔细说明了怎么根据账单核算运费.首先要确定运费规则,然后根据运费规则编写代码,生成核算列（快递费 = 省份*重量）,最后输入账单,进行核算.将脚本件生成EXE文件,就可以使用啦,需要的朋友可以参考下
    2021-05-05
  • Python如何获取当前路径并列出当前路径下的所有文件

    Python如何获取当前路径并列出当前路径下的所有文件

    这篇文章主要介绍了Python如何获取当前路径并列出当前路径下的所有文件问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
    2023-06-06
  • Python面向对象之类的定义与继承用法示例

    Python面向对象之类的定义与继承用法示例

    这篇文章主要介绍了Python面向对象之类的定义与继承用法,结合实例形式分析了Python类的定义、实例化、继承等基本操作技巧,需要的朋友可以参考下
    2019-01-01
  • Django中外键使用总结

    Django中外键使用总结

    本文主要介绍了Django中外键使用总结，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2022-07-07

最新评论