PyTorch报错：xxx.pth is a zip archive(did you mean to use torch.jit.load()?)

运行torch.load(‘xxx.pth’)报错：

RuntimeError: xxx.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或更高版本的PyTorch下，运行一下代码，来转换模型

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

参考
https://blog.csdn.net/weixin_44769214/article/details/108188126

PyTorch报错：xxx.pth is a zip archive(did you mean to use torch.jit.load()?)相关推荐

RuntimeError: xxx.pth is a zip archive (did you mean to use torch.jit.load()?)
用高版本的pytorch训练模型,在低版本中测试时出现如下错误: RuntimeError: all_ep_296.pth is a zip archive (did you mean to use ...
xxx.pth或者xxx.pt is a zip archive(did you mean to use torch.jit.load()?问题解决
提示:推荐使用方法三,不会"科学上网"的请关注微信公众号"DGX杂学"并在后台私信! 错误:使用torch.load()加载模型参数时,提示"xxx. ...
pytorch版本问题RuntimeError: params/unet.pth is a zip archive (did you mean to use torch.jit.load()?)
pytorch版本问题RuntimeError: params/unet.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 archiv ...
pytorch加载模型时出现.....ckpt_100.pth is a zip archive (did you mean to use torch.jit.load()?)
在测试加载训练好的模型时出现上方问题,参考这篇文章,原因是训练和测试的torch版本不一致. 训练的时候是1.6,测试的时候是1.2,因此需要先在1.6版本下加载模型,重新保存,在保存的时候设置use ...
【深度学习torch——error】——“xxx.pt is a zip archive(did you mean to use torch.jit.load()?)
这个问题是在进行权重文件加载进行预测的时候发生的,原因其实就是torch版本不对我是用的工作站训练使用的是torch1.7.0,然后用自己的电脑进行预测,就报错了,原因就是自己的电脑是torch1. ...
pycharm安装pytorch报错提示系列问题 torch 包找不到因为pip版本低，结果升级了pip从19.3到20.2 4又提示高版不支持torch安装
pycharm安装pytorch报错提示系列问题 torch 包找不到因为pip版本低,结果升级了pip从19.3到20.2 4又提示高版不支持torch安装 DEPRECATION: The - ...
pytorch报错Unable to get repr for
pytorch报错Unable to get repr for 标签出现了-1,2或者其他值,loss计算之后, 报错:CUDA error: device-side assert triggered ...
安装pytorch报错及解决
环境,win10,64位,Anaconda4.10.3,Cuda 9.2.148,cudnn应该为v7.4.2.24 其中在Anaconda中好像安装tensorflow-gpu 1.9 ,pytho ...
小程序-报错 xxx is not defined （已解决）
小程序-报错 xxx is not defined (已解决) 问题情境: 这样一段代码,微信的小程序报错 is not defined 我 wxml 想这样调用 //wxml 代码 <view ...

PyTorch报错：xxx.pth is a zip archive(did you mean to use torch.jit.load()?)

PyTorch报错：xxx.pth is a zip archive(did you mean to use torch.jit.load()?)相关推荐

最新文章

热门文章