toolkit 1.6版本，还是不能支持pytorch 1.6.0？

[复制链接] · 发表于 2021-1-10 17:19:49

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!

只看该作者 · 发表于 2021-1-10 18:17:31

rknntoolkit有问题请到官方rknn qq群里提问。
版本支持信息以配套文档为准。
我们还没正式发布rknn-toolkit 1.6