解决方案:
model.load_state_dict(torch.load(opt.model))
在加载模型或者权重之前加上这句
model = nn.DataParallel(model)
就可以了。
注意:
并行程序测试网络的时候需要模型也并行。单卡的模型在测试程序的也需要单卡。
解决方案:
model.load_state_dict(torch.load(opt.model))
在加载模型或者权重之前加上这句
model = nn.DataParallel(model)
就可以了。
注意:
并行程序测试网络的时候需要模型也并行。单卡的模型在测试程序的也需要单卡。