在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中sys.argv参数用法实例分析
May 20 Python
python dict.get()和dict['key']的区别详解
Jun 30 Python
深入理解 Python 中的多线程 新手必看
Nov 20 Python
windows下安装Python的XlsxWriter模块方法
May 03 Python
Python读取mat文件,并转为csv文件的实例
Jul 04 Python
使用CodeMirror实现Python3在线编辑器的示例代码
Jan 14 Python
python在回调函数中获取返回值的方法
Feb 22 Python
创建Django项目图文实例详解
Jun 06 Python
在django中,关于session的通用设置方法
Aug 06 Python
scikit-learn线性回归,多元回归,多项式回归的实现
Aug 29 Python
python-地图可视化组件folium的操作
Dec 14 Python
Python利用Turtle绘制哆啦A梦和小猪佩奇
Apr 04 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
PHPUnit PHP测试框架安装方法
2011/03/23 PHP
PHP中4种常用的抓取网络数据方法
2015/06/04 PHP
tp框架(thinkPHP)实现三次登陆密码错误之后锁定账号功能示例
2018/05/24 PHP
Javascript Jquery 遍历Json的实现代码
2010/03/31 Javascript
JS操作Cookie写入和读取实例代码
2013/10/20 Javascript
javascript中通过arguments参数伪装方法重载
2014/10/08 Javascript
Spring mvc 接收json对象
2015/12/10 Javascript
vue.js学习笔记之绑定style样式和class列表
2016/10/31 Javascript
BootStrap点击保存后实现模态框自动关闭的思路(模态框)
2017/09/26 Javascript
js实现登录与注册界面
2017/11/01 Javascript
js 实现复选框只能选择一项的示例代码
2018/01/23 Javascript
vue底部加载更多的实例代码
2018/06/29 Javascript
实例详解带参数的 npm script
2019/05/28 Javascript
小程序和web画三角形实现解析
2019/09/02 Javascript
用smtplib和email封装python发送邮件模块类分享
2014/02/17 Python
Python字符串替换实例分析
2015/05/11 Python
python编码最佳实践之总结
2016/02/14 Python
浅谈Python用QQ邮箱发送邮件时授权码的问题
2018/01/29 Python
Python格式化输出%s和%d
2018/05/07 Python
详解python while 函数及while和for的区别
2018/09/07 Python
pandas重新生成索引的方法
2018/11/06 Python
python hbase读取数据发送kafka的方法
2018/12/27 Python
利用pyshp包给shapefile文件添加字段的实例
2019/12/06 Python
python利用递归方法实现求集合的幂集
2020/09/07 Python
Nº21官方在线商店:numeroventuno.com
2019/09/26 全球购物
软件测试英文面试题
2012/10/14 面试题
历史系自荐信范文
2013/12/24 职场文书
报关员个人职业生涯规划书
2014/03/12 职场文书
司法局火灾防控方案
2014/06/05 职场文书
应用心理学专业求职信
2014/08/04 职场文书
2015出纳试用期工作总结
2014/12/12 职场文书
2015年银行个人工作总结
2015/05/14 职场文书
劳动模范获奖感言
2015/07/31 职场文书
数学备课组工作总结
2015/08/12 职场文书
银行工作心得体会范文
2016/01/23 职场文书
Oracle查看表空间使用率以及爆满解决方案详解
2022/07/23 Oracle