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 相关文章推荐
python中ConfigParse模块的用法
Sep 29 Python
python cx_Oracle的基础使用方法(连接和增删改查)
Nov 19 Python
tensorflow创建变量以及根据名称查找变量
Mar 10 Python
python3库numpy数组属性的查看方法
Apr 17 Python
Python使用sort和class实现的多级排序功能示例
Aug 15 Python
python整小时 整天时间戳获取算法示例
Feb 20 Python
如何用Python做一个微信机器人自动拉群
Jul 03 Python
python 利用pywifi模块实现连接网络破解wifi密码实时监控网络
Sep 16 Python
TensorFlow固化模型的实现操作
May 26 Python
python爬虫筛选工作实例讲解
Nov 23 Python
使用Python爬取Json数据的示例代码
Dec 07 Python
Django Paginator分页器的使用示例
Jun 23 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中一个控制字符串输出的函数
2006/10/09 PHP
模板引擎正则表达式调试小技巧
2011/07/20 PHP
php生成百度sitemap站点地图类函数实例
2014/10/17 PHP
Laravel中使用自己编写类库的3种方法
2015/02/10 PHP
Zend Framework框架路由机制代码分析
2016/03/22 PHP
PHP实现文件上传与下载
2020/08/28 PHP
一个简单的JavaScript 日期计算算法
2009/09/11 Javascript
JQuery优缺点分析说明
2011/04/10 Javascript
chrome下img加载对height()的影响示例探讨
2014/05/26 Javascript
Javascript学习笔记之 对象篇(一) : 对象的使用和属性
2014/06/24 Javascript
js用拖动滑块来控制图片大小的方法
2015/02/27 Javascript
js获取form的方法
2015/05/06 Javascript
nodejs实现遍历文件夹并统计文件大小
2015/05/28 NodeJs
js点击文本框弹出可选择的checkbox复选框
2016/02/03 Javascript
Vue.js中用webpack合并打包多个组件并实现按需加载
2017/02/17 Javascript
JS判断非空至少输入两个字符的简单实现方法
2017/06/23 Javascript
nodejs Assert中equal(),strictEqual(),deepEqual(),strictDeepEqual()比较
2017/09/18 NodeJs
js 毫秒转天时分秒的实例
2017/11/17 Javascript
ionic2.0双击返回键退出应用
2019/09/17 Javascript
Vue中keep-alive组件的深入理解
2020/08/23 Javascript
一篇文章带你从零快速上手Rollup
2020/09/07 Javascript
[03:24]2014DOTA2国际邀请赛 神秘商店生意火爆
2014/07/18 DOTA
[01:00:17]DOTA2-DPC中国联赛 正赛 SAG vs Dynasty BO3 第二场 1月25日
2021/03/11 DOTA
Go/Python/Erlang编程语言对比分析及示例代码
2018/04/23 Python
Django跨域请求CSRF的方法示例
2018/11/11 Python
python tkinter canvas 显示图片的示例
2019/06/13 Python
python 将dicom图片转换成jpg图片的实例
2020/01/13 Python
泰国办公用品购物网站:OfficeMate
2018/02/04 全球购物
法国在线宠物店:zooplus.fr
2018/02/23 全球购物
英国最大的独立玩具专卖店:The Entertainer
2019/09/06 全球购物
GWT的应用有哪两种部署模式
2012/12/21 面试题
社会实践心得体会
2014/01/03 职场文书
24岁生日感言
2014/01/13 职场文书
《火烧云》教学反思
2014/04/12 职场文书
护理学专业求职信
2014/06/29 职场文书
matplotlib之pyplot模块实现添加子图subplot的使用
2021/04/25 Python