在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制作爬虫并将抓取结果保存到excel中
Apr 06 Python
django初始化数据库的实例
May 27 Python
python机器学习之KNN分类算法
Aug 29 Python
浅谈python3.x pool.map()方法的实质
Jan 16 Python
详解Python数据分析--Pandas知识点
Mar 23 Python
Python 堆叠柱状图绘制方法
Jul 29 Python
TensorFlow 多元函数的极值实例
Feb 10 Python
浅谈python print(xx, flush = True) 全网最清晰的解释
Feb 21 Python
python GUI库图形界面开发之pyinstaller打包python程序为exe安装文件
Feb 26 Python
Python 实现绘制子图及子图刻度的变换等问题
May 31 Python
jupyter notebook保存文件默认路径更改方法汇总(亲测可以)
Jun 09 Python
Python装饰器的练习题
Nov 23 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生成WAP页面
2006/10/09 PHP
PHP+FFMPEG实现将视频自动转码成H264标准Mp4文件
2014/09/24 PHP
php计算两个坐标(经度,纬度)之间距离的方法
2015/04/17 PHP
php上传大文件失败的原因及应对策略
2015/10/20 PHP
PHP加密解密函数详解
2015/10/28 PHP
PDO::prepare讲解
2019/01/29 PHP
Javascript 文件夹选择框的两种解决方案
2009/07/01 Javascript
锋利的jQuery 要点归纳(三) jQuery中的事件和动画(上:事件篇)
2010/03/24 Javascript
为EasyUI的Tab标签添加右键菜单的方法
2012/07/14 Javascript
js实现jquery的offset()方法实例
2015/01/10 Javascript
JavaScript中Number.MIN_VALUE属性的使用示例
2015/06/04 Javascript
JS实现IE状态栏文字缩放效果代码
2015/10/24 Javascript
jQuery+css实现的tab切换标签(兼容各浏览器)
2016/01/28 Javascript
jQuery事件的绑定、触发、及监听方法简单说明
2016/05/10 Javascript
JS与Ajax Get和Post在使用上的区别实例详解
2016/06/08 Javascript
JavaScript判断微信浏览器实例代码
2016/06/13 Javascript
JavaScript DOM节点操作实例小结(新建,删除HTML元素)
2017/01/19 Javascript
JS实现获取图片大小和预览的方法完整实例【兼容IE和其它浏览器】
2017/04/24 Javascript
详解JavaScript按概率随机生成事件
2017/08/02 Javascript
Postman模拟发送带token的请求方法
2018/03/31 Javascript
详解vantUI框架在vue项目中的应用踩坑
2018/12/06 Javascript
解决cordova+vue 项目打包成APK应用遇到的问题
2019/05/10 Javascript
详解django自定义中间件处理
2018/11/21 Python
python3.6使用urllib完成下载的实例
2018/12/19 Python
selenium跳过webdriver检测并模拟登录淘宝
2019/06/12 Python
python图形用户接口实例详解
2019/12/16 Python
如何在django中运行scrapy框架
2020/04/22 Python
关于Python字符编码与二进制不得不说的一些事
2020/10/04 Python
python 如何引入协程和原理分析
2020/11/30 Python
世界知名接发和假发品牌:Poze Hair
2017/03/08 全球购物
大学生村官心得体会范文
2014/01/04 职场文书
学校万圣节活动方案
2014/02/13 职场文书
小学生寒假家长评语
2014/04/16 职场文书
党员弘扬焦裕禄精神思想汇报
2014/09/10 职场文书
网吧员工管理制度
2015/08/05 职场文书
Flask使用SQLAlchemy实现持久化数据
2021/07/16 Python