rknn-toolkit2 icon indicating copy to clipboard operation
rknn-toolkit2 copied to clipboard

rknn toolkit2中进行onnx模型转换rknn,手动修改onnx输入输出,该模型使用onnx加载推理没有问题,但是转为rknn模型load报错,输入本来是1,3,32,192,报错输入0,3,0,0

Open zhangshabao opened this issue 3 years ago • 0 comments

rknn toolkit2中进行onnx模型转换rknn,手动修改onnx输入输出,该模型使用onnx加载推理没有问题,但是转为rknn模型load报错,输入本来是1,3,32,192,报错输入0,3,0,0 请问rknn现在可以支持lstm了吗?

E load_onnx: Traceback (most recent call last): E load_onnx: File "rknn/api/rknn_base.py", line 1182, in rknn.api.rknn_base.RKNNBase.load_onnx E load_onnx: File "rknn/api/rknn_base.py", line 646, in rknn.api.rknn_base.RKNNBase._create_ir_and_inputs_meta E load_onnx: File "rknn/api/rknn_log.py", line 113, in rknn.api.rknn_log.RKNNLog.e E load_onnx: ValueError: Invalid input_shape = [0, 3, 0, 0] for input 0!

zhangshabao avatar Aug 26 '22 03:08 zhangshabao