pytorch加载模型报错Unexpected key(s) in state_dict: module.conv1.weight, module.bn1

背景

Pytorch在加载模型参数的时候，有两种情况可能出现这种问题：

1. 自己写的网络结构，例如：

• 代码

import models
arch = 'resnet50'
model = models.__dict__[arch]() # 根据resnet，自己写的的多任务网络结构
checkpoint = torch.load