1.加载全部模型:
net.load_state_dict(torch.load(net_para_pth))
2.加载部分模型
net_para_pth = './result/5826.pth'
pretrained_dict = torch.load(net_para_pth)
model_dict = net.state_dict()
pretrained_dict = {k: v for k, v in pretrained_dict.items() if k in model_dict}
model_dict.update(pretrained_dict)
net.load_state_dict(model_dict)
3.改变某一层参数后加载
将该层名称改一下,然后用2中方法加载,比如,要将conv5的out_channels由256改为512。
将conv_5改为conv_5_chg,就可以顺利加载了,不改会报错哟
算是权宜之计了,还有什么好方法,希望多多指教