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多线程编程中的join函数使用心得
Sep 02 Python
python实现根据月份和日期得到星座的方法
Mar 27 Python
python通过post提交数据的方法
May 06 Python
Python urllib、urllib2、httplib抓取网页代码实例
May 09 Python
Python通过正则表达式选取callback的方法
Jul 18 Python
Django1.9 加载通过ImageField上传的图片方法
May 25 Python
Python判断变量名是否合法的方法示例
Jan 28 Python
python删除列表元素的三种方法(remove,pop,del)
Jul 22 Python
python:批量统计xml中各类目标的数量案例
Mar 10 Python
django queryset 去重 .distinct()说明
May 19 Python
python框架flask入门之路由及简单实现方法
Jun 07 Python
PyCharm上安装Package的实现(以pandas为例)
Sep 18 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
用函数读出数据表内容放入二维数组
2006/10/09 PHP
PHP写MySQL数据 实现代码
2009/06/15 PHP
PHP中抽象类和抽象方法概念与用法分析
2016/05/24 PHP
TP5框架页面跳转样式操作示例
2020/04/05 PHP
js利用div背景,做一个竖线的效果。
2008/11/22 Javascript
Extjs学习笔记之一 初识Extjs之MessageBox
2010/01/07 Javascript
利用js实现遮罩以及弹出可移动登录窗口
2013/07/08 Javascript
用jquery.sortElements实现table排序
2014/05/04 Javascript
js判断游览器类型及版本号的代码
2014/05/11 Javascript
前端必备神器 Snap.svg 弹动效果
2014/11/10 Javascript
JS实现仿FLASH效果的竖排导航代码
2015/09/15 Javascript
Bootstrap基本样式学习笔记之表单(3)
2016/12/07 Javascript
Bootstrap基本插件学习笔记之按钮(21)
2016/12/08 Javascript
浅谈jQuery中的$.extend方法来扩展JSON对象
2017/02/12 Javascript
原生js实现移动端触摸轮播的示例代码
2017/12/22 Javascript
NodeJS 实现多语言的示例代码
2018/09/11 NodeJs
JavaScript 双向链表操作实例分析【创建、增加、查找、删除等】
2020/04/28 Javascript
微信小程序将页面按钮悬浮固定在底部的实现代码
2020/10/29 Javascript
[59:30]完美世界DOTA2联赛PWL S3 access vs LBZS 第二场 12.20
2020/12/23 DOTA
Python实现将xml导入至excel
2015/11/20 Python
Python运行不显示DOS窗口的解决方法
2018/10/22 Python
python+ffmpeg批量去视频开头的方法
2019/01/09 Python
python绘制评估优化算法性能的测试函数
2019/06/25 Python
python中PS 图像调整算法原理之亮度调整
2019/06/28 Python
浅析python内置模块collections
2019/11/15 Python
TensorFlow tensor的拼接实例
2020/01/19 Python
Keras实现DenseNet结构操作
2020/07/06 Python
英国网上自行车商店:Tredz Bikes
2019/10/29 全球购物
意大利网上药房:Farmacia 33
2020/01/27 全球购物
怎么处理XML的中文问题
2015/03/26 面试题
一套带答案的C++笔试题
2014/01/10 面试题
毕业生的自我评价分享
2013/12/18 职场文书
2015年公司新年寄语
2014/12/08 职场文书
大学生党员个人总结
2015/02/13 职场文书
python scipy 稀疏矩阵的使用说明
2021/05/26 Python
python实现简单的聊天小程序
2021/07/07 Python