使用torch.load()加载模型参数时，提示“xxx.pt is a zip archive(did you mean to use torch.jit.load()?)“

原创于 2022-04-11 20:51:12 发布 · 900 阅读

CC 4.0 BY-SA版权

文章标签：

本文讲述了如何在PyTorch 1.2老版本环境下，由于模型压缩格式导致的加载问题。通过在高版本1.6中使用torch.save设置新选项，将模型转换为非压缩格式，然后在1.2环境中成功运行。关键步骤包括使用torch.load和_save设置_use_new_zipfile_serialization=False。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题描述：

使用torch.load()加载模型参数时，提示"xxx.pt is a zip archive(did you mean to use torch.jit.load()?)"

问题原因：

代码中需要加载模型，模型原本是在高版本1.6-pytorch下加载的，属于压缩模式。

但是我的pytorch版本为1.2，属于老版本。

因此，将pytorch版本转为在1.6下，重新加载模型并将其保存为非压缩模式

_use_new_zipfile_serialization=False

然后再回到自己本身的1.2老版本pytorch，继续跑程序就可以啦，亲测好使！！！

具体代码：

新建一个.py文件，在高版本pytorch下运行以下代码（路径就是存放你的模型的路径），运行完成后，未压缩的文件就重新生成好了！！

import torch

state_dict = torch.load('checkpoints/AdaAttN/latest_net_transformer.pth', map_location="cpu")
torch.save(state_dict, 'checkpoints/AdaAttN/latest_net_transformer.pth', _use_new_zipfile_serialization=False)