Pytorch測試神經網絡時出現「RuntimeError: Error(s) in loading state_dict for Net」

解決方法：code load_state_dict(torch.load('net.pth')在前，增長 model = nn.DataParallel(model) 就能夠了。class   好比model net = NET() net.cuda() net = nn.DataParallel(net) net.load_state_dict(torch.load('net.pth')

>>阅读原文<<