pytorch load_state_dict模型，报错“Missing key(s) in state_dict、Unexpected key(s) in state_dict”

通过pytorch 平台训练一个checkpoint（xxx.pth），再infer或者finetune时，利用load_state_dict加载xxx.pth，如果报一下错误：

那么，需要检查下，在训练时，是否加入了一下语句：

net = torch.nn.DataParallel(net)

 如果，加入了以上语句，那么，在infer或者finetune时，同样需要加入以上语句，即可解决报错。