Toybrick

转化pytorch模型时报错

SodrSnne

中级会员

积分
227
发表于 2020-7-24 18:53:04    查看: 436|回复: 6 | [复制链接]    打印 | 显示全部楼层
pytorch 版本1.2.0+cpu
转化模型时遇到报错
E Catch exception when loading pytorch model: C:\Users\Administrator\Desktop\rknn-toolkit-master\fire_smoke\fire_smoke_1.pt!
Load pytorch model failed!
E Traceback (most recent call last):
E   File "rknn\api\rknn_base.py", line 611, in rknn.api.rknn_base.RKNNBase.load_pytorch
E   File "rknn\base\RKNNlib\app\importer\import_pytorch.py", line 97, in rknn.base.RKNNlib.app.importer.import_pytorch.ImportPytorch.run
E   File "rknn\base\RKNNlib\converter\convert_pytorch.py", line 552, in rknn.base.RKNNlib.converter.convert_pytorch.convert_pytorch.__init__
E   File "C:\Users\Administrator\AppData\Roaming\Python\Python36\site-packages\torch\jit\__init__.py", line 162, in load
E     cpp_module = torch._C.import_ir_module(cu, f, map_location, _extra_files)
E RuntimeError: [enforce fail at ..\caffe2\serialize\inline_container.cc:137] . PytorchStreamReader failed reading zip archive: failed finding central directory
E (no backtrace available)


跑官方pytorch 的demo是可以转化成功的

但是加载我自己的pt文件就会报这个 错误,pt文件是yolov3网络结构
https://github.com/ultralytics/yolov3  训练代码

网上搜不到相关的报错
请问该怎么解决这个问题
回复

使用道具 举报

apupu

注册会员

积分
113
发表于 2020-8-10 13:51:31 | 显示全部楼层
你好,这个问题解决了吗?
回复

使用道具 举报

leok

版主

积分
755
发表于 2020-8-11 09:00:15 | 显示全部楼层
升级到最新rknn toolkit试试。
还是解决不了的情况下,可以加QQ群1025468710,里头有rknn开发人员帮忙解答。
回复

使用道具 举报

lianbus

新手上路

积分
6
发表于 2020-8-28 13:53:07 | 显示全部楼层
我也遇到相同的问题
回复

使用道具 举报

jefferyzhang

超级版主

积分
5864
发表于 2020-8-28 14:51:41 | 显示全部楼层
回复

使用道具 举报

kkk

新手上路

积分
48
发表于 2020-9-11 09:48:07 | 显示全部楼层
楼主,问题解决了么
回复

使用道具 举报

kkk

新手上路

积分
48
发表于 2020-9-11 09:48:42 | 显示全部楼层
lianbus 发表于 2020-8-28 13:53
我也遇到相同的问题

你好,这个问题解决了么
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表