pytorch 实现打印模型的参数值
作者：布丁的自我修养 发布时间：2022-11-11 22:22:28
标签：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个参数矩阵。
来源：https://blog.csdn.net/budding0828/article/details/102646442
0
投稿猜你喜欢
题目给定一个字符串，逐个翻转字符串中的每个单词。示例 1：输入: "the sky is blue"输出: "- 本文实例讲述了wxPython中listbox用法。分享给大家供大家参考。具体如下：# load a listbox with names,
- HTML5 越来越引起人们的关注，苹果甚至将 HTML5 视为 Flash 的掘墓人 。然而，作为一种尚未成型的技术，HTML5 对很多人来
- 什么是浅克隆、深克隆浅克隆：直接将存储在栈中的值赋值给对应变量，如果是基本数据类型，则直接赋值对应的值，如果是引用类型，则赋值的是地址。深克
- 最近发现数据库服务器压力很大，CPU经常达到100％。查看进程，发现有大量的sp_cursorclose;1进程信息。网上查了下，出现sp_
- 通过界面设计上是能手工操作的，无法达到我批量修改几千台服务器。 因为此了一个脚本来批量执行。 环境：redgate + mssql 2008
- 本文实例讲述了python实现的接收邮件功能。分享给大家供大家参考，具体如下：一 简介本代码实现从网易POP3服务器接收邮件二 代码impo
- 在NumPy中，矩阵是 ndarray 的子类，与数学概念中的矩阵一样，NumPy中的矩阵也是二维的，可以使用 mat 、 matrix 以
- 前言yield的英文单词意思是生产，刚接触Python的时候感到非常困惑，一直没弄明白yield的用法。最近又重新学习了下，所以整理了下面这
- 第一步，下载PHPphp官网地址windows 下载直接解压即可liunx请自行csdn搜索教程第二步，下载code插件1. PHP Deb
- 如果点击“保存文本文件”将会弹出保存文本文件的对话框口，保存的文本文件中的内容是如下：/* 2010年4月5日0
- 在有些情况下，利用try…except来捕捉异常可以起到代替if…else的作用。比如在判断一个链表是否存在环的leetcode题目中，初始
- 如何使用migrations的使用非常简单: 修改model, 比如增加field, 然后运行python manager.py makem
- 二分类或分类问题，网络输出为二维矩阵：批次x几分类，最大的为当前分类，标签为one-hot型的二维矩阵：批次x几分类计算百分比有numpy和
- 在开发的过程中，经常会遇到需要给别人提供一个输入框，然后别人输入一些ID，或者关键字的东西，例如wordpress的后台的标签输入框：这个就
- 本文介绍的圆角方法很特别，有创意。昨天群里大家讨论到一个豆瓣上的按钮，下面这个。在这里用到了不同宽度的按钮，但是只是用到了一个图片。&nbs
- Python Json使用本篇主要介绍一下 python 中 json的使用 如何把 dict转成json 、object 转成json 、
- 从Access数据库中选取记录有件最令人丧气的事情，它们是以怎样的顺序输入到数据库内就按照怎样的顺序出来。就算你在Access环境内采用So
- 一、获取二叉树的深度就是二叉树最后的层次，如下图：实现代码：def getheight(self): &n
- binlog二进制日志对于mysql数据库的重要性有多大，在此就不多说了。下面根据本人的日常操作经历，并结合网上参考资料，对binlog日志
