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编写脚本获取手机当前应用apk的信息
Jul 21 Python
介绍Python中的文档测试模块
Apr 28 Python
python实现应用程序在右键菜单中添加打开方式功能
Jan 09 Python
Python科学画图代码分享
Nov 29 Python
TensorFlow 滑动平均的示例代码
Jun 19 Python
Django 实现购物车功能的示例代码
Oct 08 Python
python PyAutoGUI 模拟鼠标键盘操作和截屏功能
Aug 04 Python
Python3 requests文件下载 期间显示文件信息和下载进度代码实例
Aug 16 Python
python 使用opencv 把视频分割成图片示例
Dec 12 Python
pytorch之添加BN的实现
Jan 06 Python
Python爬虫教程之利用正则表达式匹配网页内容
Dec 08 Python
Python时间操作之pytz模块使用详解
Jun 14 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.ini-dist 和 php.ini-recommended 的区别介绍(方便开发与安全的朋友)
2012/07/01 PHP
php引用计数器进行垃圾收集机制介绍
2012/09/19 PHP
php接口和抽象类使用示例详解
2014/03/02 PHP
php中fsockopen用法实例
2015/01/05 PHP
php实现随机显示图片方法汇总
2015/05/21 PHP
php实现给一张图片加上水印效果
2016/01/02 PHP
PHP连续签到功能实现方法详解
2019/12/04 PHP
thinkphp框架表单数组实现图片批量上传功能示例
2020/04/04 PHP
js的闭包的一个示例说明
2008/11/18 Javascript
javascript 设计模式之单体模式 面向对象学习基础
2010/04/18 Javascript
javascript 获取iframe里页面中元素值的方法
2014/02/17 Javascript
超链接的禁用属性Disabled使用示例
2014/07/31 Javascript
jQuery中closest()函数用法实例
2015/01/07 Javascript
js+HTML5实现视频截图的方法
2015/06/16 Javascript
JS实现的文字与图片定时切换效果代码
2015/10/06 Javascript
javascript动态生成树形菜单的方法
2015/11/14 Javascript
AngularJS页面访问时出现页面闪烁问题的解决
2016/03/06 Javascript
JQuery实现DIV其他动画效果的简单实例
2016/09/18 Javascript
layui数据表格 table.render 报错的解决方法
2019/09/29 Javascript
vue实现输入一位数字转汉字功能
2019/12/13 Javascript
[03:39]这就是刀塔,我们是冠军!燃情短片讲述我们的DOTA故事
2019/07/02 DOTA
Python中字典映射类型的学习教程
2015/08/20 Python
Django使用Celery异步任务队列的使用
2018/03/13 Python
Python如何获取文件指定行的内容
2020/05/27 Python
使用python实现名片管理系统
2020/06/18 Python
澳大利亚首屈一指的在线购物目的地:Kogan.com
2017/02/02 全球购物
Radley英国官网:英国莱德利小狗包
2019/03/21 全球购物
银行实习鉴定
2013/12/13 职场文书
大学校庆策划书
2014/01/31 职场文书
环保志愿者活动总结
2014/06/27 职场文书
2014年学校总务处工作总结
2014/12/08 职场文书
Python文件的操作示例的详细讲解
2021/04/08 Python
python3 实现mysql数据库连接池的示例代码
2021/04/17 Python
聊聊JS ES6中的解构
2021/04/29 Javascript
新手入门Mysql--概念
2021/06/18 MySQL
经典《舰娘》游改全新动画预告 预定11月开播
2022/04/01 日漫