print输出 pytorch_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 实现打印模型的参数值就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持云海天教程。

print输出 pytorch_pytorch 实现打印模型的参数值相关推荐

print输出 pytorch_pytorch打印网络结构的实例
最简单的方法当然可以直接print(net),但是这样网络比较复杂的时候效果不太好,看着比较乱:以前使用caffe的时候有一个网站可以在线生成网络框图,tensorflow可以用tensor boar ...
PyTorch 打印模型结构、输出维度和参数信息（torchsummary）
使用 PyTorch 深度学习搭建模型后,如果想查看模型结构,可以直接使用 print(model) 函数打印.但该输出结果不是特别直观,查阅发现有个能输出类似 keras 风格 model.summ ...
Python中的标准库函数(内置函数)print()输出(打印出)字符串的常见用法
这篇博文用于记录下Python中的标准库函数print()的常见用法,随着时间的推移,可能会有更新. print 在 Python3.x 是一个函数,但在 Python2.x 版本不是一个函数,只是一 ...
python3打印不换行加逗号_python3让print输出不换行的方法
python3让print输出不换行的方法 python 3.x版本print输出不换行的格式如下: print(x, end="") 其中,end="" 可使 ...
python中怎么输出中文-python中使用print输出中文的方法
看Python简明教程,学习使用print打印字符串,试了下打印中文,不行. 编辑环境:IDLE 上网搜了下解决办法,各种说法,试了两种: print u"学习" print (u ...
python取中间值的函数_tensorflow实现在函数中用tf.Print输出中间值
tensorflow由于其基于静态图的模式,导致写代码的时候很难调试,除了用官方的调试工具外,最直接的方法就是把中间结果输出出来查看,然而,直接用print函数只能输出tensor变量的形状,而不是数 ...
python2中文输出代码_解决vscode python print 输出窗口中文乱码的问题
解决vscode python print 输出窗口中文乱码的问题发布时间:2020-09-17 23:53:25 来源:脚本之家阅读:119 一.搭建 python 环境在 VSC 中点击 F ...
pythonprint中文教程_python中使用print输出中文的方法
看Python简明教程,学习使用print打印字符串,试了下打印中文,不行. 编辑环境:IDLE 上网搜了下解决办法,各种说法,试了两种: print u"学习" print (u ...
python不换行空格输出_解决Python print输出不换行没空格的问题
解决Python print输出不换行没空格的问题今天在做编程题的时候发现Python的print输出默认换行输出,并且输出后有空格. 题目要求输出 122 而我的输出是: 1 2 2 于是我百度查 ...

print输出 pytorch_pytorch 实现打印模型的参数值

print输出 pytorch_pytorch 实现打印模型的参数值相关推荐

最新文章

热门文章