Toybrick

onnx模型推理正常,转换rknn时报错 算子 shape异常

shinima

新手上路

积分
36
楼主
发表于 3 天前    查看: 105|回复: 0 | [复制链接]    打印 | 只看该作者
错误具体内容如下,但使用onnx推理时并没有该错误发生:
Traceback (most recent call last):
  File "convert_work_dir/tools/model-transfer-tool/./onnx2rknn_visual_prompt_model.py", line 99, in main
    ret = rknn.build(do_quantization=False)
  File "/root/miniconda3/envs/py39/lib/python3.9/site-packages/rknn/api/rknn.py", line 192, in build
    return self.rknn_base.build(do_quantization=do_quantization, dataset=dataset, expand_batch_size=rknn_batch_size)
  File "rknn/api/rknn_log.py", line 349, in rknn.api.rknn_log.error_catch_decorator.error_catch_wrapper
  File "rknn/api/rknn_log.py", line 95, in rknn.api.rknn_log.RKNNLog.e
ValueError: Traceback (most recent call last):
  File "rknn/api/rknn_log.py", line 344, in rknn.api.rknn_log.error_catch_decorator.error_catch_wrapper
  File "rknn/api/rknn_base.py", line 1945, in rknn.api.rknn_base.RKNNBase.build
  File "rknn/api/graph_optimizer.py", line 1013, in rknn.api.graph_optimizer.GraphOptimizer.fold_constant
  File "rknn/api/rknn_log.py", line 95, in rknn.api.rknn_log.RKNNLog.e
ValueError: The shape of '/encoder/Cast_3_output_0' is (0,), it is abnormal!

回复

使用道具 举报

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

本版积分规则

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


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