pytorch低版本环境加载高版本（1.6+）模型报错（xxx is a zip archive (did you mean to use torch.jit.load()?)）

SUN_SU3

于 2021-03-17 14:58:22 发布

阅读量5.7k

点赞数 7

分类专栏：模型文章标签：深度学习

本文链接：https://blog.csdn.net/u013546508/article/details/114930772

版权

模型专栏收录该内容

20 篇文章 1 订阅

订阅专栏

问题描述

transformers 4.0+版本训练出来的模型，在3.0的Model.from_pretrained中加载会报错，pytorch_model.bin is a zip archive (did you mean to use torch.jit.load()?)
这是因为pytorch1.6及之后,更换了保存模型文件的方式

解决方法

将高版本的，模型文件重新保存一下，加上参数_use_new_zipfile_serialization=False，这样模型就可以在1.6以下的环境中运行了

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