零点教学网(www.0djx.com)-学电脑,学软件,学编程,学网站,学设计!

pytorch获取模型某一层参数名及参数值

零点教学网 Python教程 2020-09-24 00:38:51 7
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])


零点教学网,,自学EXCEL、自学PS、自学CAD、自学C语言、自学css3实例,是一个通过网络自主学习工作技能的自学平台,网友喜欢的软件自学网站。

如果以上内容你喜欢,请持续关注:零点教学网软件编程教程栏目

分享:

本文来自投稿,不代表本人立场,如若转载,请注明出处:http://0djx.com/ruanjianbianchengjiaocheng/python/54677.html

  python编程入门电子版   pytorch获取模型某一层参数名及参数值  

(7)
打赏 微信扫一扫
« 上一篇 2020年09月24日 00:38:23
下一篇 » 2020年09月24日 00:39:09

python编程入门电子版_相关内容

pytorch获取模型某一层参数名及参数值_相关内容