使用pytorch的并行测试网络的时候报错: RuntimeError: Error(s) in loading state_dict for DataParallel

解决方案:
model.load_state_dict(torch.load(opt.model))
在加载模型或者权重之前加上这句
model = nn.DataParallel(model)
就可以了。

注意：
并行程序测试网络的时候需要模型也并行。单卡的模型在测试程序的也需要单卡。