pytorch加载模型时出现.....xxx.pth is a zip archive (did you mean to use torch.jit.load()?)

最新推荐文章于 2024-04-20 23:40:42 发布

沫小希

最新推荐文章于 2024-04-20 23:40:42 发布

阅读量791

点赞数 1

分类专栏：深度学习文章标签： pytorch 深度学习人工智能

本文链接：https://blog.csdn.net/qq_43581644/article/details/125331965

版权

深度学习专栏收录该内容

11 篇文章 0 订阅

订阅专栏

这是因为pytorch1.6以后默认把模型存储为压缩文件，而老版本不是压缩文件，因此识别不了
解决方法：可以在新版本中将模型加载后，再存储为非压缩文件，再用老版本加载：

    def save(self, apath, epoch, is_best=False):
        target = self.get_model()
        torch.save(
            target.state_dict(), 
            os.path.join(apath, 'model', 'model_latest.pt'),
            _use_new_zipfile_serialization=False
        )
        if is_best:
            torch.save(
                target.state_dict(),
                os.path.join(apath, 'model', 'model_best.pt'),
                _use_new_zipfile_serialization=False
            )