修改参数值
方法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(<