Toybrick

标题: onnx模型推理正常,转换rknn时报错 算子 shape异常 [打印本页]

作者: shinima    时间: 3 天前
标题: onnx模型推理正常,转换rknn时报错 算子 shape异常
错误具体内容如下,但使用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!






欢迎光临 Toybrick (https://t.rock-chips.com/) Powered by Discuz! X3.3