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程序实例(通讯录)
Nov 29 Python
Python标准库urllib2的一些使用细节总结
Mar 16 Python
Python中random模块生成随机数详解
Mar 10 Python
Python 中urls.py:URL dispatcher(路由配置文件)详解
Mar 24 Python
Pipenv一键搭建python虚拟环境的方法
May 22 Python
python random从集合中随机选择元素的方法
Jan 23 Python
Python实现通过解析域名获取ip地址的方法分析
May 17 Python
Python实现某论坛自动签到功能
Aug 20 Python
Python zip函数打包元素实例解析
Dec 11 Python
Python类反射机制使用实例解析
Dec 30 Python
经验丰富程序员才知道的8种高级Python技巧
Jul 27 Python
python 窃取摄像头照片的实现示例
Jan 08 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实现的功能是显示8条基色色带
2006/10/09 PHP
php文档更新介绍
2011/07/22 PHP
thinkphp5 加载静态资源路径与常量的方法
2017/12/24 PHP
JQUERY复选框CHECKBOX全选,取消全选
2008/08/30 Javascript
jQuery ajax 路由和过滤器使用说明
2011/08/02 Javascript
JS 添加网页桌面快捷方式的代码详细整理
2012/12/27 Javascript
js判断游览器类型及版本号的代码
2014/05/11 Javascript
图文详解JavaScript的原型对象及原型链
2016/08/02 Javascript
详解vue2.0组件通信各种情况总结与实例分析
2017/03/22 Javascript
Angular CLI 安装和使用教程
2017/09/13 Javascript
微信小程序Flex布局用法深入浅出分析
2019/04/25 Javascript
微信小程序 wx:for 与 wx:for-items 与 wx:key的正确用法
2020/05/19 Javascript
[15:58]DOTA2国际邀请赛采访专栏:Tongfu.Sansheng&KingJ,DK.rOtk
2013/08/08 DOTA
python连接mysql数据库示例(做增删改操作)
2013/12/31 Python
Python中IPYTHON入门实例
2015/05/11 Python
python处理二进制数据的方法
2015/06/03 Python
python创建列表并给列表赋初始值的方法
2015/07/28 Python
详解Python网络爬虫功能的基本写法
2016/01/28 Python
Python编程对列表中字典元素进行排序的方法详解
2017/05/26 Python
浅谈用VSCode写python的正确姿势
2017/12/16 Python
Python面向对象class类属性及子类用法分析
2018/02/02 Python
Flask框架Flask-Principal基本用法实例分析
2018/07/23 Python
Python对接六大主流数据库(只需三步)
2019/07/31 Python
通过字符串导入 Python 模块的方法详解
2019/10/27 Python
Python中Flask-RESTful编写API接口(小白入门)
2019/12/11 Python
python中的 zip函数详解及用法举例
2020/02/16 Python
利用python对excel中一列的时间数据更改格式操作
2020/07/14 Python
Python 处理日期时间的Arrow库使用
2020/08/18 Python
python3列表删除大量重复元素remove()方法的问题详解
2021/01/04 Python
Python使用cn2an实现中文数字与阿拉伯数字的相互转换
2021/03/02 Python
怀俄明州飞钓:Platte River Fly Shop
2017/12/28 全球购物
介绍一下.NET构架下remoting和webservice
2014/05/08 面试题
售后服务经理岗位职责范本
2014/02/22 职场文书
学习党的群众路线教育实践活动心得体会
2014/03/01 职场文书
新闻传播专业求职信
2014/07/22 职场文书
博士导师推荐信
2015/03/25 职场文书