pytorch获取模型某一层参数名及参数值方式
2020-02-13 11:32
1566 查看
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获取模型某一层参数名及参数值方式就是小编分享给大家的全部内容了,希望能给大家一个参考
您可能感兴趣的文章:
相关文章推荐
- pytorch获取模型某一层参数名及参数值
- tensorflow获取预训练模型某层参数并赋值到当前网络指定层方式
- Pytorch中实现只导入部分模型参数的方式
- js方法获取get方式传过来的参数(不乱码)
- NodeJs获取请求参数的方法(get方式)
- spring mvc获取路径参数的几种方式 - 浅夏的个人空间 - 开源中国社区
- ajax get方式 调用 web api的方法 如何获取参数
- struts2访问ServletAPI方式和获取参数的方式
- Nodejs express 获取url参数,post参数的三种方式
- spring Aop 配置文件方式+JoinPoint获取参数
- 通过内存模型深入理解参数传递方式
- ireport实践之request获取参数名和参数值的两种方式
- 前台序列化传过来的值,后台获取之后封装到map当中,让后在转化成json格式,最后在把json里面的参数里面的某一个值进行分割,最后在存到json格式的数据中去。
- AJAX POST请求中参数以form data和request payload形式在servlet中的获取方式
- 获取url参数的几种方式
- asp.net 通过ajax方式调用webmethod方法使用自定义类传参及获取返回参数
- HttpServletRequest request 获取form参数的两种方式
- Struts2获取参数的三种方式
- struts2 接受参数的两种方式(属性驱动和模型驱动)
- 表单中多个请求参数名字相同,服务器到底获取的是哪个请求参数的值,表单的action和get和post提交方式的关系以及提交数据的不同点