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

问题描述：

使用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)