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进来就可以了