错误处理:RuntimeError: [enforce fail at ..\caffe2\serialize\inline_container.cc:145] . PytorchStreamRead
生活随笔
收集整理的这篇文章主要介绍了
错误处理:RuntimeError: [enforce fail at ..\caffe2\serialize\inline_container.cc:145] . PytorchStreamRead
小编觉得挺不错的,现在分享给大家,帮大家做个参考.
pytorch加载程序的时候,出现以下问题
resnet = timm.create_model('densenet121',num_classes=1000,pretrained=True)model=torch.nn.Sequential(resnet,torch.nn.Linear(1000,500),torch.nn.Tanh(),torch.nn.Linear(500,100),torch.nn.Tanh(),torch.nn.Linear(100,26))
net3=model
net3.load_state_dict(torch.load('densenet18-10500.pkl', map_location='cpu'))
net3
RuntimeError: [enforce fail at ..\caffe2\serialize\inline_container.cc:145] . PytorchStreamReader failed reading zip archive: failed finding central directory
出错的原因是pkl文件损坏
换成正确的pkl文件,load进来就可以了
总结
以上是生活随笔为你收集整理的错误处理:RuntimeError: [enforce fail at ..\caffe2\serialize\inline_container.cc:145] . PytorchStreamRead的全部内容,希望文章能够帮你解决所遇到的问题。
- 上一篇: PyTorch笔记: GPU上训练的模型
- 下一篇: 错误处理:安装torch-sparse、