pytorch 实现打印模型的参数值


Posted in Python onDecember 30, 2019

对于简单的网络

例如全连接层Linear

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

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

输出如下:

pytorch 实现打印模型的参数值

由于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])

输出:

pytorch 实现打印模型的参数值

pytorch 实现打印模型的参数值

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

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

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

Python 相关文章推荐
python实现2014火车票查询代码分享
Jan 10 Python
Python合并两个字典的常用方法与效率比较
Jun 17 Python
Python开发的HTTP库requests详解
Aug 29 Python
详解 Python 与文件对象共事的实例
Sep 11 Python
Python数据分析中Groupby用法之通过字典或Series进行分组的实例
Dec 08 Python
解析Python的缩进规则的使用
Jan 16 Python
在Python中通过getattr获取对象引用的方法
Jan 21 Python
PyTorch: 梯度下降及反向传播的实例详解
Aug 20 Python
jupyter notebook 调用环境中的Keras或者pytorch教程
Apr 14 Python
python如何修改文件时间属性
Feb 05 Python
Pytorch中的学习率衰减及其用法详解
Jun 05 Python
全网非常详细的pytest配置文件
Jul 15 Python
Python如何基于smtplib发不同格式的邮件
Dec 30 #Python
pytorch获取模型某一层参数名及参数值方式
Dec 30 #Python
Python类反射机制使用实例解析
Dec 30 #Python
Python读取YAML文件过程详解
Dec 30 #Python
python通过nmap扫描在线设备并尝试AAA登录(实例代码)
Dec 30 #Python
pytorch中获取模型input/output shape实例
Dec 30 #Python
Python读取csv文件实例解析
Dec 30 #Python
You might like
PHP 线程安全与非线程安全版本的区别深入解析
2013/08/06 PHP
php5.3 不支持 session_register() 此函数已启用的解决方法
2013/11/12 PHP
Linux下安装oracle客户端并配置php5.3
2014/10/12 PHP
JS对象与JSON格式数据相互转换
2012/02/20 Javascript
js事件(Event)知识整理
2012/10/11 Javascript
多种方法实现360浏览器下禁止自动填写用户名密码
2014/06/16 Javascript
使用documentElement正确取得当前可见区域的大小
2014/07/25 Javascript
原生JavaScript生成GUID的实现示例
2014/09/05 Javascript
javascript批量修改文件编码格式的方法
2015/01/27 Javascript
JavaScript实现对下拉列表值进行排序的方法
2015/07/15 Javascript
JavaScript小技巧整理篇(非常全)
2016/01/26 Javascript
简介EasyUI datagrid editor combogrid搜索框的实现
2016/04/01 Javascript
jQuery防止重复绑定事件的解决方法
2016/05/14 Javascript
利用原生js和jQuery实现单选框的勾选和取消操作的方法
2016/09/04 Javascript
Vue开发过程中遇到的疑惑知识点总结
2017/01/20 Javascript
Javascript将图片的绝对路径转换为base64编码的方法
2018/01/11 Javascript
Vue CLI3.0中使用jQuery和Bootstrap的方法
2019/02/28 jQuery
Vue 使用Props属性实现父子组件的动态传值详解
2019/11/13 Javascript
jquery使用echarts实现有向图可视化功能示例
2019/11/25 jQuery
Map与WeakMap类型在JavaScript中的使用详解
2020/11/18 Javascript
[52:32]完美世界DOTA2联赛PWL S2 Magma vs LBZS 第三场 11.18
2020/11/18 DOTA
Python新手们容易犯的几个错误总结
2017/04/01 Python
python实现键盘控制鼠标移动
2020/11/27 Python
浅谈python str.format与制表符\t关于中文对齐的细节问题
2019/01/14 Python
python随机生成大小写字母数字混合密码(仅20行代码)
2020/02/01 Python
欧舒丹澳洲版:L’OCCITANE
2017/07/17 全球购物
新加坡领先的时尚生活方式零售品牌:CHARLES & KEITH
2018/01/16 全球购物
Java程序员面试题
2016/09/27 面试题
short s1 = 1; s1 = s1 + 1;有什么错? short s1 = 1; s1 += 1;有什么错?
2014/09/26 面试题
运动会邀请函范文
2014/02/06 职场文书
公司委托书范本
2014/04/04 职场文书
2014年劳动部工作总结
2014/12/11 职场文书
廉洁自律承诺书2015
2015/01/22 职场文书
2015年食品安全工作总结
2015/05/15 职场文书
离婚案件上诉状
2015/05/23 职场文书
护士岗位竞聘书
2015/09/15 职场文书