分类：测试神经网络

2021
07-22

Pytorch测试神经网络时出现 RuntimeError:的解决方案

Pytorch测试神经网络时出现“RuntimeError:Error(s)inloadingstate_dictforNet”解决方法：load_state_dict(torch.load('net.pth')在前，增加model=nn.DataParallel(model)就可以了。比如net=NET()net.cuda()net=nn.DataParallel(net)net.load_state_dict(torch.load('net.pth')补充：解决RuntimeError:Error(s)inloadingstate_dictforXXXX在运行代码时遇到了这个错误，显示错误对应代码中的state_dict，找到对... 继续阅读 >