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 实现打印模型的参数值就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
SQLAlchemy 是 Python 生态系统中非常流行的数据库处理库,它提供了一种高效、简洁的方式与数据库进行交互,在数据驱动的应用程序中,复杂查询是必不可少的,本文将通过一些常见的示例介绍如何使用 SQLAlchemy 编写复杂查询,需要的朋友可以参考下2024-10-10
在本篇文章里小编给大家分享的是关于Python中无限循环的条件的相关文章，需要的朋友们可以参考下。2020-05-05
这篇文章主要介绍了jupyter notebook的安装与使用详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-05-05
.pkl文件是Python中用于存储对象的文件格式,全称是"pickle",它是Python标准库中的一个模块,用于将Python对象序列化,以便于在不同的Python程序之间进行传输或存储,这篇文章主要给大家介绍了如何通过一文详细介绍Python中pkl格式文件的相关资料,需要的朋友可以参考下2024-05-05
在本篇内容里小编给大家整理了关于python3中TQDM库安装及使用的相关知识点，有兴趣的朋友们可以学习下。2020-11-11
这篇文章主要介绍了Python bsonrpc源码的相关资料，帮助大家更好的理解和学习使用python，感兴趣的朋友可以了解下2021-03-03
本文主要介绍了python主动抛出异常raise的方法实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2023-12-12
这篇文章主要为大家分析了python正则表达式常见面试题，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-01-01
本文总结了如何获得SSL证书并给Django项目配置HTTPS，建议先收藏再阅读，将来有一天你很可能会用到它。2021-04-04
这篇文章主要介绍了python中对开区间和闭区间的理解，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-07-07
最新评论