.pth is a zip archive (did you mean to use torch.jit.load()?)

本文主要是介绍.pth is a zip archive (did you mean to use torch.jit.load()?)，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

RuntimeError: bert.pth is a zip archive (did you mean to use torch.jit.load()?)
 

.pth is a zip archive (did you mean to use torch.jit.load()?)

原因
xxx.pth来自pytorch1.6或更高的版本。1.6之后pytorch默认使用zip文件格式来保存权重文件，导致这些权重文件无法直接被1.5及以下的pytorch加载。

解决方案
在pytorch1.6版本下运行：

state_dict = torch.load("xxx.pth")
torch.save(state_dict, "xxx.pth", _use_new_zipfile_serialization=False)
1
2

https://blog.csdn.net/song_wheaver/article/details/112527697

 

这篇关于.pth is a zip archive (did you mean to use torch.jit.load()?)的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---