分类：加载预训练模型

2021
07-01

pytorch加载预训练模型与自己模型不匹配的解决方案

pytorch中如果自己搭建网络并且加载别人的与训练模型的话，如果模型和参数不严格匹配，就可能会出问题，接下来记录一下我的解决方法。两个有序字典找不同模型的参数和pth文件的参数都是有序字典(OrderedDict)，把字典中的键转为列表就可以在for循环里迭代找不同了。model=ResNet18(1)model_dict1=torch.load('resnet18.pth')model_dict2=model.state_dict()model_list1=list(model_dict1.keys())model_list2=list(model_... 继续阅读 >