pytorch 实现打印模型的参数值

更新时间：2019年12月30日 09:34:32 作者：布丁的自我修养

今天小编就为大家分享一篇pytorch 实现打印模型的参数值，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

对于简单的网络

例如全连接层Linear

可以使用以下方法打印linear层：

fc = nn.Linear(3, 5)
params = list(fc.named_parameters())
print(params.__len__())
print(params[0])
print(params[1])

输出如下：

由于Linear默认是偏置bias的，所有参数列表的长度是2。第一个存的是全连接矩阵，第二个存的是偏置。

对于稍微复杂的网络

例如MLP

mlp = nn.Sequential(
      nn.Dropout(p=0.3),
      nn.Linear(1024, 256),
      nn.Linear(256, 64),
      nn.Linear(64, 16),
      nn.Linear(16, 1)
    )
params = list(mlp.named_parameters())
print(params.__len__())

print(params[0])
print(params[1])

print(params[2])
print(params[3])

输出：

可以发现，堆叠起来的网络，参数是依次放置的。先是全连接的权重，然后偏置。然后是下一层网络的权重+偏置。依次进行下去。

这里有4层fc，4*2=8.所以一共有8个参数矩阵。

以上这篇pytorch 实现打印模型的参数值就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

Python读写unicode文件的方法
这篇文章主要介绍了Python读写unicode文件的方法,涉及Python针对文件的读取及编码操作的相关技巧,具有一定参考借鉴价值,需要的朋友可以参考下
2015-07-07
Python更换pip源方法过程解析
这篇文章主要介绍了Python更换pip源方法过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2020-05-05
python 中文字符串的处理实现代码
python中直接将一个中文字符串赋值给一个变量使用的是utf-8格式的编码,以下为证.
2009-10-10
Python脚本传参数argparse模块的使用
这篇文章主要介绍了Python脚本传参数argparse模块的使用，文章围绕argparse模块的相关资料展开具体的使用方法，具有一的参考价值，需要的小伙伴可以参考一下
2022-03-03
Python sklearn分类决策树方法详解
决策树(Decision Tree）是在已知各种情况发生概率的基础上，通过构成决策树来求取净现值的期望值大于等于零的概率，评价项目风险，判断其可行性的决策分析方法，是直观运用概率分析的一种图解法
2022-09-09
Python3爬虫中识别图形验证码的实例讲解
在本篇内容里小编给大家分享的是关于Python3爬虫中识别图形验证码的实例讲解内容，需要的朋友们可以学习参考下。
2020-07-07
详谈pandas中agg函数和apply函数的区别
下面小编就为大家分享一篇详谈pandas中agg函数和apply函数的区别，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-04-04
PyMongo安装使用笔记
这篇文章主要介绍了PyMongo安装使用笔记,本文讲解使用pip方式安装PyMongo,并给出一个pymongo的简单使用代码实例,需要的朋友可以参考下
2015-04-04
python里dict变成list实例方法
在本篇内容里小编给大家分析了关于python里dict变成list实例方法的实例内容，对此有需要的朋友们可以参考学习下。
2019-06-06
Python+Appium实现自动化清理微信僵尸好友的方法
这篇文章主要介绍了Python+Appium实现自动化清理微信僵尸好友的方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2021-02-02

pytorch 实现打印模型的参数值

相关文章

最新评论

大家感兴趣的内容

最近更新的内容

常用在线小工具