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基于checksum计算文件是否相同的方法
Jul 09 Python
Python自动化开发学习之三级菜单制作
Jul 14 Python
使用Python进行目录的对比方法
Nov 01 Python
使用Rasterio读取栅格数据的实例讲解
Nov 26 Python
django框架中ajax的使用及避开CSRF 验证的方式详解
Dec 11 Python
在echarts中图例legend和坐标系grid实现左右布局实例
May 16 Python
在keras里实现自定义上采样层
Jun 28 Python
django models里数据表插入数据id自增操作
Jul 15 Python
Python中实现一行拆多行和多行并一行的示例代码
Sep 06 Python
python制作一个简单的gui 数据库查询界面
Nov 19 Python
Python列表删除重复元素与图像相似度判断及删除实例代码
May 07 Python
PYTHON 使用 Pandas 删除某列指定值所在的行
Apr 28 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单例模式示例分享
2015/02/12 PHP
PHP编程获取各个时间段具体时间的方法
2017/05/26 PHP
PHP与JavaScript针对Cookie的读写、交互操作方法详解
2017/08/07 PHP
PHP whois查询类定义与用法示例
2019/04/03 PHP
JQuery AJAX实现目录浏览与编辑的代码
2008/10/21 Javascript
说明你的Javascript技术很烂的五个原因
2011/04/26 Javascript
jQuery插件原来如此简单 jQuery插件的机制及实战
2012/02/07 Javascript
JS 退出系统并跳转到登录界面的实现代码
2013/06/29 Javascript
JavaScript制作简易的微信打飞机
2015/03/31 Javascript
jQuery实现的文字hover颜色渐变效果实例
2016/02/20 Javascript
Angular.js中$apply()和$digest()的深入理解
2016/10/13 Javascript
彻底学会Angular.js中的transclusion
2017/03/12 Javascript
详解Vue中使用v-for语句抛出错误的解决方案
2017/05/04 Javascript
js生成word中图片处理方法
2018/01/06 Javascript
讲解vue-router之命名路由和命名视图
2018/05/28 Javascript
Vue和React组件之间的传值方式详解
2019/01/31 Javascript
jQuery实现条件搜索查询、实时取值及升降序排序的方法分析
2019/05/04 jQuery
Jquery异步上传文件代码实例
2019/11/13 jQuery
解决VUE-Router 同一页面第二次进入不刷新的问题
2020/07/22 Javascript
vue+elementUI 实现内容区域高度自适应的示例
2020/09/26 Javascript
python实现在无须过多援引的情况下创建字典的方法
2014/09/25 Python
Python入门篇之字符串
2014/10/17 Python
python数据结构链表之单向链表(实例讲解)
2017/07/25 Python
Python从数据库读取大量数据批量写入文件的方法
2018/12/10 Python
python监测当前联网状态并连接的实例
2018/12/18 Python
Django中的静态文件管理过程解析
2019/08/01 Python
详解Ubuntu环境下部署Django+uwsgi+nginx总结
2020/04/02 Python
英国最大的邮寄种子和植物公司:Thompson & Morgan
2017/09/21 全球购物
沙特阿拉伯家用电器和电子产品购物网站:Sheta and Saif
2020/04/03 全球购物
Java面试中常遇到的问题,也是需要注意的几点
2013/08/30 面试题
高中打架检讨书
2014/02/13 职场文书
中学生评语大全
2014/04/18 职场文书
纪念九一八事变演讲稿:青少年应树立远大理想
2014/09/14 职场文书
介绍长城的导游词
2015/01/30 职场文书
python某漫画app逆向
2021/03/31 Python
python 镜像环境搭建总结
2022/09/23 Python