Pytroch网络模型：修改参数值，修改参数名，添加参数层，删除参数层

HuiYu-Li

于 2019-08-22 22:49:31 发布

阅读量1.2w

点赞数 14

CC 4.0 BY-SA版权

分类专栏： Pytorch

本文链接：https://blog.csdn.net/weixin_44058333/article/details/99682848

修改参数值

方法1

dict的类型是collecitons.OrderedDict，是一个有序字典，直接将新参数名称和初始值作为键值对插入，然后保存即可。

#修改前
dict = torch.load('./ckpt_dir//model_0.pth')
net.load_state_dict(dict)
for name,param in net.named_parameters():
	print(name,param)
#按参数名修改权重
dict["forward1.0.weight"] = torch.ones((1,1,3,3,3))
dict["forward1.0.bias"] = torch.ones(