加载模型出现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)

你可能感兴趣的:(配置环境和小工具专栏,深度学习,人工智能,计算机视觉)