遇到这种错误的时候,找了很久,发现原来是服务器pytorch版本和笔记本版本不一样,那就重新将模型保存一边:
checkpoint = "*.pth"
new_checkpoint = "new*.pth"
model = net()
model.load_state_dict(torch.load(checkpoint))
model.eval()
torch.save(model.state_dict(), new_checkpoint, use_new_zipfile_serialization=False)