加载模型出现in load_state_dict raise RuntimeError(‘Error(s) in loading state_dict

报错(RuntimeError: Error(s) in loading state_dict for Model)详细信息如下：

原因：自己的模型训练的时候使用了torch.nn.DataParallel实现多卡同时训练，加载训练好的模型进行测试的时候没使用torch.nn.DataParallel就会报这样的错误。

解决方式：
在测试的代码中对模型使用下行代码

model = torch.nn.DataParallel(model)