Toybrick

toolkit 1.6版本,还是不能支持pytorch 1.6.0?

nj1tony

中级会员

积分
228
发表于 2021-1-10 17:19:49    查看: 7665|回复: 1 | [复制链接]    打印 | 显示全部楼层
load模型的时候出错。
python3 test.py
--> config model
done
--> Loading model
./yolov5s-3.pt ********************
E Catch exception when loading pytorch model: ./yolov5s-3.pt!
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 339, in rknn.api.rknn_base.RKNNBase.load_pytorch
E   File "rknn/base/RKNNlib/RK_nn.py", line 146, in rknn.base.RKNNlib.RK_nn.RKnn.load_pytorch
E   File "rknn/base/RKNNlib/app/importer/import_pytorch.py", line 128, in rknn.base.RKNNlib.app.importer.import_pytorch.ImportPytorch.run
E   File "rknn/base/RKNNlib/converter/convert_pytorch_new.py", line 2243, in rknn.base.RKNNlib.converter.convert_pytorch_new.convert_pytorch.load
E   File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/torch/jit/__init__.py", line 275, in load
E     cpp_module = torch._C.import_ir_module(cu, f, map_location, _extra_files)
E RuntimeError: [enforce fail at inline_container.cc:144] . PytorchStreamReader failed reading zip archive: failed finding central directory
Load pytorch model failed!



回复

使用道具 举报

jefferyzhang

版主

积分
12844
发表于 2021-1-10 18:17:31 | 显示全部楼层
rknntoolkit有问题请到官方rknn qq群里提问。
版本支持信息以配套文档为准。
我们还没正式发布rknn-toolkit 1.6
回复

使用道具 举报

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

本版积分规则

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


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