pytorch获取模型某一层参数名及参数值方式


Posted in Python onDecember 30, 2019

1、Motivation:

I wanna modify the value of some param;

I wanna check the value of some param.

The needed function:

2、state_dict() #generator type

model.modules()#generator type

named_parameters()#OrderDict type

from torch import nn
import torch
#creat a simple model
model = nn.Sequential(
  nn.Conv3d(1,16,kernel_size=1),
  nn.Conv3d(16,2,kernel_size=1))#tend to print the W of this layer
input = torch.randn([1,1,16,256,256])
if torch.cuda.is_available():
  print('cuda is avaliable')
  model.cuda()
  input = input.cuda()
#打印某一层的参数名
for name in model.state_dict():
  print(name)
#Then I konw that the name of target layer is '1.weight'

#schemem1(recommended)
print(model.state_dict()['1.weight'])

#scheme2
params = list(model.named_parameters())#get the index by debuging
print(params[2][0])#name
print(params[2][1].data)#data

#scheme3
params = {}#change the tpye of 'generator' into dict
for name,param in model.named_parameters():
params[name] = param.detach().cpu().numpy()
print(params['0.weight'])

#scheme4
for layer in model.modules():
if(isinstance(layer,nn.Conv3d)):
  print(layer.weight)

#打印每一层的参数名和参数值
#schemem1(recommended)
for name,param in model.named_parameters():
  print(name,param)

#scheme2
for name in model.state_dict():
  print(name)
  print(model.state_dict()[name])

以上这篇pytorch获取模型某一层参数名及参数值方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
pyqt4教程之实现windows窗口小示例分享
Mar 07 Python
python实现从web抓取文档的方法
Sep 26 Python
让 python 命令行也可以自动补全
Nov 30 Python
在Python中进行自动化单元测试的教程
Apr 15 Python
Python匹配中文的正则表达式
May 11 Python
Python中正则表达式详解
May 17 Python
Python实现的根据IP地址计算子网掩码位数功能示例
May 23 Python
numpy基础教程之np.linalg
Feb 12 Python
python numpy生成等差数列、等比数列的实例
Feb 25 Python
如何在python中执行另一个py文件
Apr 30 Python
Pytorch实现将模型的所有参数的梯度清0
Jun 24 Python
python glom模块的使用简介
Apr 13 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
Pytorch Tensor的统计属性实例讲解
Dec 30 #Python
PyTorch中permute的用法详解
Dec 30 #Python
You might like
php 之 没有mysql支持时的替代方案
2006/10/09 PHP
PHP连接SQLServer2005方法及代码
2013/12/26 PHP
php 中htmlentities导致中文无法查询问题
2018/09/10 PHP
php中的依赖注入实例详解
2019/08/14 PHP
php设计模式之备忘模式分析【星际争霸游戏案例】
2020/03/24 PHP
日期函数扩展类Ver0.1.1
2006/09/07 Javascript
如何实现JS函数的重载
2006/09/22 Javascript
Mootools 1.2教程 Tooltips
2009/09/15 Javascript
jQuery右键菜单contextMenu使用实例
2011/09/28 Javascript
Jquery Uploadify上传带进度条的简单实例
2014/02/12 Javascript
使用jquery实现放大镜效果
2014/09/02 Javascript
学习javascript面向对象 理解javascript原型和原型链
2016/01/04 Javascript
Vue.js组件tab实现选项卡切换
2020/03/23 Javascript
原生JS实现N级菜单的代码
2017/05/21 Javascript
Vue.js 中的 $watch使用方法
2017/05/25 Javascript
selenium 反爬虫之跳过淘宝滑块验证功能的实现代码
2020/08/27 Javascript
[40:31]Secret vs Alliacne 2019国际邀请赛小组赛 BO2 第二场 8.15
2019/08/17 DOTA
详解Python中的Cookie模块使用
2015/07/06 Python
机器学习python实战之决策树
2017/11/01 Python
python使用pycharm环境调用opencv库
2018/02/11 Python
Python tkinter的grid布局及Text动态显示方法
2018/10/11 Python
Python openpyxl读取单元格字体颜色过程解析
2019/09/03 Python
python中bytes和str类型的区别
2019/10/21 Python
python 爬取英雄联盟皮肤并下载的示例
2020/12/04 Python
python读取excel数据并且画图的实现示例
2021/02/08 Python
台湾菁英交友:结识黄金单身的台湾人
2018/01/22 全球购物
中国电子产品批发商/跨境电商/外贸网:Sunsky-online
2020/04/20 全球购物
推荐信格式要求
2014/05/09 职场文书
演讲稿格式范文
2014/05/19 职场文书
大学生优秀班干部事迹材料
2014/05/26 职场文书
消防安全标语
2014/06/07 职场文书
红领巾广播站广播稿
2015/08/19 职场文书
2016年公司中秋节致辞
2015/11/26 职场文书
深度学习详解之初试机器学习
2021/04/14 Python
Python中X[:,0]和X[:,1]的用法
2021/05/10 Python
5种 JavaScript 方式实现数组扁平化
2021/10/05 Javascript