错误处理: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进来就可以了

你可能感兴趣的:(其他,html)