pytorch加載程序的時候，出現(xiàn)以下問題 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的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。