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

原因

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

解决方案

在含有高版本的torch的设备上运行如下代码：

import torch
state_dict = torch.load("**.pth",map_location='cpu')
torch.save(state_dict, "**.pth", _use_new_zipfile_serialization=False)