【KeyError: ‘unexpected key “module.*****“】【KeyError: ‘unexpected key “net.“】

下载的预训练模型显示key不对，
可能是原模型使用pytorch的并行训练nn.DataParallel，存储时套了一层net，还喜欢在key前面加module，可以替换掉：

import torch
param_path = 'pretrained/****'
new_dict =  {}
pretrained_dict = torch.load(param_path, map_location=torch.device('cpu'))
pretrained_dict = pretrained_dict['net']
for k, v in pretrained_dict.items():
    k = k.replace("module.", '') 
    new_dict[k] = v
torch.save(new_dict, params_path)