错误处理: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

Python zipfile:RuntimeError:文件密码错误

RuntimeError:cuda 运行时错误(710):设备端断言触发于