在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中的多线程编程
Apr 09 Python
python实现获取Ip归属地等信息
Aug 27 Python
pycharm 实现显示project 选项卡的方法
Jan 17 Python
python3实现钉钉消息推送的方法示例
Mar 14 Python
Python3 实现串口两进程同时读写
Jun 12 Python
在cmd中查看python的安装路径方法
Jul 03 Python
python通过SSH登陆linux并操作的实现
Oct 10 Python
Python3 sys.argv[ ]用法详解
Oct 24 Python
解决Python列表字符不区分大小写的问题
Dec 19 Python
TensorFlow查看输入节点和输出节点名称方式
Jan 04 Python
python实现代码审查自动回复消息
Feb 01 Python
python中super()函数的理解与基本使用
Aug 30 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使用内置dir类实现目录遍历删除
2015/03/31 PHP
php的常量和变量实例详解
2017/06/27 PHP
关于jquery动态增减控件的一些想法和小插件
2010/08/01 Javascript
鼠标拖动实现DIV排序示例代码
2013/10/14 Javascript
jquery如何扑捉回车键触发的事件
2014/04/24 Javascript
node.js中的fs.close方法使用说明
2014/12/17 Javascript
jQuery中slice()方法用法实例
2015/01/07 Javascript
text-align:justify实现文本两端对齐 兼容IE
2015/08/19 Javascript
基于jQuery实现的无刷新表格分页实例
2016/02/17 Javascript
jQuery的事件预绑定
2016/12/05 Javascript
深入讲解xhr(XMLHttpRequest)/jsonp请求之abort
2017/07/26 Javascript
AngularJS 将再发布一个重要版本 然后进入长期支持阶段
2018/01/31 Javascript
微信小程序 点击切换样式scroll-view实现代码实例
2019/10/11 Javascript
Python基础之循环语句用法示例【for、while循环】
2019/03/23 Python
基于Python中random.sample()的替代方案
2020/05/23 Python
Html5定位终极解决方案
2020/02/05 HTML / CSS
联想C++笔试题
2012/06/13 面试题
某公司.Net方向面试题
2014/04/24 面试题
公司会计岗位职责
2014/02/13 职场文书
给老师的一封建议书
2014/03/13 职场文书
幼师求职自荐信
2014/05/31 职场文书
学校献爱心活动总结
2014/07/08 职场文书
中职三好学生事迹材料
2014/08/24 职场文书
商铺门面租房协议书
2014/10/21 职场文书
敬业奉献模范事迹材料
2014/12/24 职场文书
初中作文评语
2014/12/25 职场文书
工作检讨书范文
2015/01/23 职场文书
仓库保管员岗位职责
2015/02/09 职场文书
个人党性分析总结
2015/03/05 职场文书
2015毕业实习推荐信
2015/03/23 职场文书
生产现场禁烟通知
2015/04/23 职场文书
新年寄语2016
2015/08/17 职场文书
2015年六年级班主任工作总结
2015/10/15 职场文书
《最后一头战象》教学反思
2016/02/16 职场文书
mysql 联合索引生效的条件及索引失效的条件
2021/11/20 MySQL
Win11筛选键导致键盘失灵怎么解决? Win11关闭筛选键的技巧
2022/04/08 数码科技