pytorch之打印网络节点

1 输入：

import torchthe_model = torch.load("model.pt")
print (the_model)params=the_model.state_dict()
len(params)
for i, j in enumerate(params):print i, j

2 输出：

RNNModel((drop): Dropout(p=0.2)(encoder): Embedding(33278, 200)(rnn): LSTM(200, 200, num_layers=2, dropout=0.2)(decoder): Linear(in_features=200, out_features=33278, bias=True)
)0 encoder.weight
1 rnn.weight_ih_l0
2 rnn.weight_hh_l0
3 rnn.bias_ih_l0
4 rnn.bias_hh_l0
5 rnn.weight_ih_l1
6 rnn.weight_hh_l1
7 rnn.bias_ih_l1
8 rnn.bias_hh_l1
9 decoder.weight
10 decoder.bias

3 当然，使用model.named_parameters()也能打印出网络参数和名称，有兴趣的可以试试啊。

pytorch之打印网络节点相关推荐

pytorch 打印网络参数
import utils VGG = networks.VGG19('vgg19.pth', feature_mode=True) VGG.to(device) VGG.eval() print('- ...
pytorch贝叶斯网络_贝叶斯神经网络：2个在TensorFlow和Pytorch中完全连接
pytorch贝叶斯网络贝叶斯神经网络 (Bayesian Neural Net) This chapter continues the series on Bayesian deep learni ...
＜计算机视觉四＞ pytorch版yolov3网络搭建
鼠标点击下载项目源代码免费下载地址 <计算机视觉一> 使用标定工具标定自己的目标检测 <计算机视觉二> labelme标定的数据转换成yolo训练格式 <计算机 ...
利用Pytorch实现GoogLeNet网络
目录 1 GoogLeNet网络 1.1 网络结构及参数 1.2 Inception结构 1.3 带降维功能的Inception结构 1.4 辅助分类器 2 利用Pytorch实现GoogLeNe ...
Pytorch搭建LeNet5网络
本讲目标: 介绍Pytorch搭建LeNet5网络的流程. Pytorch八股法搭建LeNet5网络 1.LeNet5网络介绍 2.Pytorch搭建LeNet5网络 2.1搭建LeNet网络 2 ...
Pytorch搭建FCN网络
Pytorch搭建FCN网络前言原理代码实现前言 FCN 全卷积网络,用卷积层替代CNN的全连接层,最后通过转置卷积层得到一个和输入尺寸一致的预测结果: 原理为了得到更好的分割结果,论文中提 ...
pytorch空间变换网络
pytorch空间变换网络本文将学习如何使用称为空间变换器网络的视觉注意机制来扩充网络.可以在DeepMind paper 有关空间变换器网络的内容. 空间变换器网络是对任何空间变换的差异化关注的概 ...
R语言使用igraph包进行网络（network）可视化实战：将变量映射到网络节点（vertex）、将变量映射到网络的边（edge）
R语言使用igraph包进行网络(network)可视化实战:将变量映射到网络节点(vertex).将变量映射到网络的边(edge) 目录
php打印从1到1000,输入一个链表的头节点，从尾到头打印每个节点的值，利用PHP如何实现？...
以下是我的实现代码(我只是打印出了每个节点): /** * Question:输入一个链表的头节点,反向打印链表节点 * Author:entner * time: 2017-7-21 * versi ...

pytorch之打印网络节点

pytorch之打印网络节点相关推荐

最新文章

热门文章