‘torch.load‘ received a zip file that looks like a TorchScript

于 2023-10-08 22:40:24 发布
阅读量465 收藏 2
点赞数 2
文章标签： python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/m0_73585512/article/details/133691526
版权

复现代码时出现如下bug:

原因：

经过百度后发现是版本问题，model-path来自pytorch1.6或更高的版本。1.6之后pytorch默认使用zip文件格式来保存权重文件，导致这些权重文件无法直接被1.5及以下的pytorch加载。

解决方案：

将原代码“torch.load() " 改为 "torch.jit.load" 

问题解决

drink_sing
关注
  • 2
    点赞
  • 踩
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
torch.load received a zip file
jacke121的专栏
04-09 3066
torch.load received a zip file cltt torch.load()加载模型，提示xxx.pt is a zip archive(did you mean to use torch.jit.load()?) 参考链接 https://blog.csdn.net/irober/article/details/115144522?utm_medium=distribute.pc_relevant.none-task-blog-baidujs_title-0&sp.
yolov5s.torchscript.pt
10-13
yolov5 v3.0版本，根据yolov5s.pt权重处理的，适合在移动端推理的torchscript序列化文件。
【Pytorch】使用torch.load()加载模型时报错，提示加载的模型是zip格式
qq_44846512的博客
07-12 2233
问题 在pytorch版本1.6.0上保存最优模型，当将该模型下载到其他设备上时发现模型从.pt转成了zip压缩包格式或者直接使用在其他版本的pytorch上加载时报错。 解决方法 torch官网上指出 The 1.6 release of PyTorch switched torch.save to use a new zipfile-based file format. torch.load still retains the ability to load files in the old forma
使用torch.load()加载模型参数时，提示“xxx.pt is a zip archive(did you mean to use torch.jit.load()?)“
qq_35037684的博客
12-08 4946
参考博客： https://blog.csdn.net/weixin_44769214/article/details/108188126 用方法1解决了问题
xxx.pth或者xxx.pt is a zip archive(did you mean to use torch.jit.load()?问题解决
DG息的博客
10-14 2915
错误：使用torch.load()加载模型参数时，提示“xxx.pt is a zip archive(did you mean to use torch.jit.load()?)“或者“xxx.pth is a zip archive(did you mean to use torch.jit.load()?)“ 错误原因： xxx.pth或者xxx.pt来自pytorch1.6或更高的版本。1.6之后pytorch默认使用zip文件格式来保存权重文件，导致这些权重文件无法直接被1.5及以下的pytorc
PyTorch 源码解读之 torch.serialization & torch.hub
zandaoguang的博客
10-28 4067
作者 | 123456来源 | OpenMMLab编辑 | 极市平台导读本文解读基于PyTorch 1.7版本，对torch.serialization、torch.save和tor...
torch1.7.zip
01-26
标题中的"torch1.7.zip"表明这是一个包含PyTorch版本1.7的压缩包，特别为树莓派4B优化。PyTorch是一个流行的开源机器学习库，它提供了动态计算图功能，使得构建和训练神经网络变得简单易行。在Python环境中，它常被...
PyTorch的torch.cat用法
09-16
PyTorch中的`torch.cat`函数是一个非常重要的操作，它用于将多个张量沿着指定的维度进行拼接。这个函数在构建复杂的神经网络模型，特别是在处理图像数据和序列数据时非常常用。下面我们将详细探讨`torch.cat`的使用...
基于torch.where和布尔索引的速度比较
09-18
今天小编就为大家分享一篇基于torch.where和布尔索引的速度比较，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
Pytorch学习（十七）--- 模型load各种问题解决
热门推荐
Hungryof的专栏
08-02 7万+
简单的模型load 一般来说，保存模型是把参数全部用model.cpu().state_dict(), 然后加载模型时一般用 model.load_state_dict(torch.load(model_path))。 值得注意的是：torch.load 返回的是一个 OrderedDict. import torch import torch.nn as nn class Net_old(...
Pytorch框架TorchScript模型转换方法
胖胖大海的博客
04-20 7638
为什么要使用TorchScript对模型进行转换？ a)、TorchScript代码可以在它自己的解释器中调用，它本质上是一个受限的Python解释器。这个解释器不获取全局解释器锁，因此可以在同一个实例上同时处理多个请求。 b)、这种格式允许我们将整个模型保存到磁盘上，并将其加载到另一个环境中，比如用Python以外的语言编写的服务器中 c)、TorchScript提供了一种表示方式，我们可以在其中对代码进行编译器优化，以提供更有效的执行 d)、TorchScript允许我们与许多后端/设备运行时进
torch.load加载模型后出现问题
qq_39511822的博客
06-04 1359
torch.load后模型效果非常差
【解决方法】torch.load()加载模型参数报错“xxx.pth is a zip archive(did you mean to use torch.jit.load()?)“
song_wheaver的博客
01-12 1万+
问题表现 运行torch.load("xxx.pth")加载模型参数，报错： RuntimeError: xxx.pth is a zip archive(did you mean to use torch.jit.load()?) 原因 xxx.pth来自pytorch1.6或更高的版本。1.6之后pytorch默认使用zip文件格式来保存权重文件，导致这些权重文件无法直接被1.5及以下的pytorch加载。 解决方案 在pytorch1.6版本下运行： state_dict = torch.load(
Pytorch C++部署 之 TorchScript 踩坑记录
ll_cookie的博客
04-04 1万+
小白一枚，最近尝试用c++调pytorch模型，网上搜了一对资料，大同小异，都是需要先将模型文件转换成torch script。 pytorch官网提供两种方式，1：trace 2：注释。 这两种方式怎么操作，官网以及很多博客都有详细介绍，不细扣，我也扣不明白，毕竟我是个小白，放链接供参考： https://blog.csdn.net/dou3516/article/details/829124...
pytorch使用过程中遇到的警告
legalhighhigh的博客
11-04 1万+
1、UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use tensor.item() to convert a 0-dim tensor to a Python number #原语句： train_loss+=loss.data[0] #修改后： train_lo...
Pytorch学习笔记（二）使用Pytorch的常见错误汇总
ROM的博客
07-31 7万+
那些年我们一起踩过的坑 _, pred = output.data.cpu().topk(1, dim=1) RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorCopy.c:70...
torch.load和torch.load可以连接吗
最新发布
12-11
根据提供的引用内容，torch.save()和torch.load()分别用于保存和加载PyTorch模型。它们是独立的命令，不能直接连接在一起使用。torch.save()用于将模型保存到文件中，而torch.load()用于从文件中加载模型。如果要在不同的程序中使用已保存的模型，则需要在新程序中使用torch.load()加载该模型。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值