在pytorch中查看可训练参数的例子


Posted in Python onAugust 18, 2019

pytorch中我们有时候可能需要设定某些变量是参与训练的,这时候就需要查看哪些是可训练参数,以确定这些设置是成功的。

pytorch中model.parameters()函数定义如下:

def parameters(self):
    r"""Returns an iterator over module parameters.

    This is typically passed to an optimizer.

    Yields:
      Parameter: module parameter

    Example::

      >>> for param in model.parameters():
      >>>   print(type(param.data), param.size())
      <class 'torch.FloatTensor'> (20L,)
      <class 'torch.FloatTensor'> (20L, 1L, 5L, 5L)

    """
    for name, param in self.named_parameters():
      yield param

所以,我们可以遍历named_parameters()中的所有的参数,只打印那些param.requires_grad=True的变量。具体实现代码如下所示:

for name, param in model.named_parameters():
  if param.requires_grad:
    print(name)

这样打印出的结果就是模型中所有的可训练参数列表!

以上这篇在pytorch中查看可训练参数的例子就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
独特的python循环语句
Nov 20 Python
Python+Wordpress制作小说站
Apr 14 Python
批量获取及验证HTTP代理的Python脚本
Apr 23 Python
Python解析json之ValueError: Expecting property name enclosed in double quotes: line 1 column 2(char 1)
Jul 06 Python
Python操作mongodb数据库进行模糊查询操作示例
Jun 09 Python
对python中的 os.mkdir和os.mkdirs详解
Oct 16 Python
创建Django项目图文实例详解
Jun 06 Python
Django使用中间键实现csrf认证详解
Jul 22 Python
Python3实现配置文件差异对比脚本
Nov 18 Python
Python 面向对象部分知识点小结
Mar 09 Python
Python生成器generator原理及用法解析
Jul 20 Python
Python基于字典实现switch case函数调用
Jul 22 Python
浅析PyTorch中nn.Module的使用
Aug 18 #Python
关于PyTorch 自动求导机制详解
Aug 18 #Python
pytorch神经网络之卷积层与全连接层参数的设置方法
Aug 18 #Python
pytorch numpy list类型之间的相互转换实例
Aug 18 #Python
对Pytorch中nn.ModuleList 和 nn.Sequential详解
Aug 18 #Python
pytorch 自定义数据集加载方法
Aug 18 #Python
PyTorch的Optimizer训练工具的实现
Aug 18 #Python
You might like
php数组函数序列之array_unique() - 去除数组中重复的元素值
2011/10/29 PHP
php实现mysql事务处理的方法
2014/12/25 PHP
php通过递归方式复制目录和子目录的方法
2015/03/13 PHP
javascript 获取url参数和script标签中获取url参数函数代码
2010/01/22 Javascript
Jquery实现图片放大镜效果的思路及代码(自写)
2013/10/18 Javascript
jquery修改网页背景颜色通过css方法实现
2014/06/06 Javascript
兼容Firefox的Javascript XSLT 处理XML文件
2014/12/31 Javascript
移动端 一个简单易懂的弹出框
2016/07/06 Javascript
JS简单生成两个数字之间随机数的方法
2016/08/03 Javascript
JS 实现Base64编码与解码实例详解
2016/11/07 Javascript
jquery select2的使用心得(推荐)
2016/12/04 Javascript
深入理解Node module模块
2018/03/26 Javascript
Vue项目从webpack3.x升级webpack4不完全指南
2019/04/28 Javascript
js针对图片加载失败的处理方法分析
2019/08/24 Javascript
Vue 实现简易多行滚动&quot;弹幕&quot;效果
2020/01/02 Javascript
微信小程序实现搜索框功能及踩过的坑
2020/06/19 Javascript
Vue 解决父组件跳转子路由后当前导航active样式消失问题
2020/07/21 Javascript
Python自动化构建工具scons使用入门笔记
2015/03/10 Python
详解python 发送邮件实例代码
2016/12/22 Python
Python实现感知器模型、两层神经网络
2017/12/19 Python
Python基于property实现类的特性操作示例
2018/06/15 Python
python3中zip()函数使用详解
2018/06/29 Python
对python mayavi三维绘图的实现详解
2019/01/08 Python
Python进阶之全面解读高级特性之切片
2019/02/19 Python
利用PyCharm Profile分析异步爬虫效率详解
2019/05/08 Python
python3 写一个WAV音频文件播放器的代码
2019/09/27 Python
Django框架序列化与反序列化操作详解
2019/11/01 Python
python颜色随机生成器的实例代码
2020/01/10 Python
CSS3的一个简单导航栏实现
2015/08/03 HTML / CSS
Sam’s Club山姆会员商店:沃尔玛旗下高端会员制商店
2017/01/16 全球购物
留学生如何写好自荐信
2013/12/27 职场文书
新学期班主任寄语
2014/01/18 职场文书
办公用房租赁协议书
2014/11/29 职场文书
企业开业庆典答谢词
2015/01/20 职场文书
幼儿园秋季开学通知
2015/07/16 职场文书
MySQL下使用Inplace和Online方式创建索引的教程
2021/05/26 MySQL