错误处理:RuntimeError: [enforce fail at ..caffe2serializeinline_container.cc:145] . PytorchStreamRead(代码
Posted UQI-LIUWJ
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了错误处理:RuntimeError: [enforce fail at ..caffe2serializeinline_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 ..caffe2serializeinline_container.cc:145] . PytorchStreamRead(代码的主要内容,如果未能解决你的问题,请参考以下文章
错误处理:RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be
Discord.py - 重新启动命令错误 - RuntimeError:事件循环已关闭
在PyQt脚本中使用多处理时的RuntimeError和IOError
RuntimeError:预期 1D 目标张量,不支持多目标 Python:NumPy