同一个模型，使用nn.DataParallel前后，pytorch加载参数文件时报错：missing key(s) in state_dict

问题描述

使用nn.DataParallel训练并保存的pt文件，使用单卡推理并加载预训练权重时，显示缺失一部分的key。如下所示：
图1：加载预训练权重的语句。

图2.1：报错信息，显示缺失一部分的key。

图2.2 同时，报错信息显示多出了一部分key：

原因分析和解决方案：

可能因为使用了并行计算，参数文件的key命名不一致，将图1的语句修改如下，也就是推理时模型也使用DataParallel封装：