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文件夹与文件的操作实现代码
Jul 13 Python
pygame学习笔记(1):矩形、圆型画图实例
Apr 15 Python
使用简单工厂模式来进行Python的设计模式编程
Mar 01 Python
Python基础教程之正则表达式基本语法以及re模块
Mar 25 Python
深入浅析ImageMagick命令执行漏洞
Oct 11 Python
Python基于动态规划算法解决01背包问题实例
Dec 06 Python
Python实现抓取HTML网页并以PDF文件形式保存的方法
May 08 Python
基于PyQt4和PySide实现输入对话框效果
Feb 27 Python
微信小程序python用户认证的实现
Jul 29 Python
flask框架自定义过滤器示例【markdown文件读取和展示功能】
Nov 08 Python
在tensorflow中设置保存checkpoint的最大数量实例
Jan 21 Python
Python tcp传输代码实例解析
Mar 18 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时间戳使用实例代码
2008/06/07 PHP
phpmyadmin 3.4 空密码登录的实现方法
2010/05/29 PHP
PHP的基本常识小结
2013/07/05 PHP
php读取excel文件示例分享(更新修改excel)
2014/02/27 PHP
php使用curl存储cookie的示例
2014/03/31 PHP
thinkphp5使用无限极分类
2019/02/18 PHP
php设计模式之工厂模式用法经典实例分析
2019/09/20 PHP
laravel 解决多库下的DB::transaction()事务失效问题
2019/10/21 PHP
基于PHP实现生成随机水印图片
2020/12/09 PHP
PHP常用字符串输出方法分析(echo,print,printf及sprintf)
2021/03/09 PHP
document.getElementBy("id")与$("#id")有什么区别
2013/09/22 Javascript
jquery动态调整div大小使其宽度始终为浏览器宽度
2014/06/06 Javascript
webpack 2的react开发配置实例代码
2017/07/28 Javascript
解决在Bootstrap模糊框中使用WebUploader的问题
2018/03/22 Javascript
vue.js+element-ui动态配置菜单的实例
2018/09/07 Javascript
原生js实现贪食蛇小游戏的思路详解
2019/11/26 Javascript
js中关于Blob对象的介绍与使用
2019/11/29 Javascript
JS document对象简单用法完整示例
2020/01/14 Javascript
在js文件中引入(调用)另一个js文件的三种方法
2020/09/11 Javascript
在Python中通过threading模块定义和调用线程的方法
2016/07/12 Python
Django objects.all()、objects.get()与objects.filter()之间的区别介绍
2017/06/12 Python
使用实现XlsxWriter创建Excel文件并编辑
2018/05/04 Python
tensorflow 只恢复部分模型参数的实例
2020/01/06 Python
python pandas利用fillna方法实现部分自动填充功能
2020/03/16 Python
联想法国官方网站:Lenovo法国
2018/10/18 全球购物
益模软件Java笔试题
2012/03/27 面试题
介绍一下Linux中的链接
2016/06/05 面试题
数学与统计学院学生个人职业生涯规划书
2014/02/10 职场文书
优秀少先队员主要事迹材料
2014/05/28 职场文书
服务理念标语
2014/06/18 职场文书
长城导游词300字
2015/01/30 职场文书
布达拉宫导游词
2015/02/02 职场文书
2015年环卫工作总结
2015/04/28 职场文书
保险公司反洗钱宣传活动总结
2015/05/08 职场文书
民政局未婚证明
2015/06/15 职场文书
62句有关感恩节文案(推荐收藏)
2019/11/28 职场文书