Toybrick

关于onnx转rknn时loading model失败的问题

Chambers

新手上路

积分
40
楼主
发表于 2020-5-25 17:48:43 | 显示全部楼层
hello,我自己训练了一个pytorch-efficientNet,转换为onnx以后,onnx2rknn遇到类似的错误,请教一下,如何解决的?
回复

使用道具 举报

Chambers

新手上路

积分
40
沙发
发表于 2020-5-25 17:51:52 | 显示全部楼层
报错如下:
  * https://github.com/tensorflow/addons
If you depend on functionality not listed there, please file an issue.

/home/air/.local/lib/python3.6/site-packages/onnx_tf/common/__init__.py:87: UserWarning: FrontendHandler.get_outputs_names is deprecated. It will be removed in future release.. Use node.outputs instead.
  warnings.warn(message)
--> config model
done
D Save log info to: ./e.log
E Catch exception when loading onnx model: efficient-b0-140.onnx!
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 343, in rknn.api.rknn_base.RKNNBase.load_onnx
E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 523, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.__init__
E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 527, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.__init__
E   File "rknn/base/RKNNlib/converter/onnx_util.py", line 154, in rknn.base.RKNNlib.converter.onnx_util.ONNXProto_Util.optim_model
E   File "/home/air/.local/lib/python3.6/site-packages/onnx/optimizer.py", line 55, in optimize
E     optimized_model_str = C.optimize(model_str, passes)
E IndexError: _Map_base::at
Load model failed!
回复

使用道具 举报

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

本版积分规则

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


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