|
rknn-toolkit 版本:1.4.0
转换路径:caffe模型转rknn
报错日志:
E Catch exception when building RKNN model!
E Traceback (most recent call last):
E File "/home/users/luozhiwen/miniconda3/envs/py3.6/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1659, in _create_c_op
E c_op = c_api.TF_FinishOperation(op_desc)
E tensorflow.python.framework.errors_impl.InvalidArgumentError: Dimensions must be equal, but are 56 and 256 for 'elementwise_mul_20_elementwise_mul_0_17_2/elementwise_mul_20_elementwise_mul_0_17' (op: 'Mul') with input shapes: [1,256,56,56], [1,256].
E During handling of the above exception, another exception occurred:
E Traceback (most recent call last):
E File "rknn/api/rknn_base.py", line 895, in rknn.api.rknn_base.RKNNBase.build
E File "rknn/api/rknn_base.py", line 1797, in rknn.api.rknn_base.RKNNBase._quantize2
E File "rknn/base/RKNNlib/app/medusa/quantization.py", line 105, in rknn.base.RKNNlib.app.medusa.quantization.Quantization.run
E File "rknn/base/RKNNlib/app/medusa/quantization.py", line 44, in rknn.base.RKNNlib.app.medusa.quantization.Quantization._run_quantization
E File "rknn/base/RKNNlib/app/medusa/workspace.py", line 139, in rknn.base.RKNNlib.app.medusa.workspace.Workspace.run
E File "rknn/base/RKNNlib/app/medusa/workspace.py", line 109, in rknn.base.RKNNlib.app.medusa.workspace.Workspace._setup_graph
E File "rknn/base/RKNNlib/app/medusa/workspace.py", line 110, in rknn.base.RKNNlib.app.medusa.workspace.Workspace._setup_graph
E File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 274, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build
E File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 278, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build
E File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 305, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 305, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 305, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E [Previous line repeated 39 more times]
E File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 331, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 336, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E File "rknn/base/RKNNlib/layer/RKNNlayer.py", line 287, in rknn.base.RKNNlib.layer.RKNNlayer.RKNNLayer.compute_tensor
E File "rknn/base/RKNNlib/layer/multiplylayer.py", line 80, in rknn.base.RKNNlib.layer.multiplylayer.MultiplyLayer.compute_out_tensor
E File "/home/users/luozhiwen/miniconda3/envs/py3.6/lib/python3.6/site-packages/tensorflow/python/util/dispatch.py", line 180, in wrapper
E return target(*args, **kwargs)
E File "/home/users/luozhiwen/miniconda3/envs/py3.6/lib/python3.6/site-packages/tensorflow/python/ops/math_ops.py", line 248, in multiply
E return gen_math_ops.mul(x, y, name)
E File "/home/users/luozhiwen/miniconda3/envs/py3.6/lib/python3.6/site-packages/tensorflow/python/ops/gen_math_ops.py", line 5860, in mul
E "Mul", x=x, y=y, name=name)
E File "/home/users/luozhiwen/miniconda3/envs/py3.6/lib/python3.6/site-packages/tensorflow/python/framework/op_def_library.py", line 788, in _apply_op_helper
E op_def=op_def)
E File "/home/users/luozhiwen/miniconda3/envs/py3.6/lib/python3.6/site-packages/tensorflow/python/util/deprecation.py", line 507, in new_func
E return func(*args, **kwargs)
E File "/home/users/luozhiwen/miniconda3/envs/py3.6/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 3300, in create_op
E op_def=op_def)
E File "/home/users/luozhiwen/miniconda3/envs/py3.6/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1823, in __init__
E control_input_ops)
E File "/home/users/luozhiwen/miniconda3/envs/py3.6/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1662, in _create_c_op
E raise ValueError(str(e))
E ValueError: Dimensions must be equal, but are 56 and 256 for 'elementwise_mul_20_elementwise_mul_0_17_2/elementwise_mul_20_elementwise_mul_0_17' (op: 'Mul') with input shapes: [1,256,56,56], [1,256].
Build model failed!
问题:想请问一下elementwise mul算子目前还不支持broadcast功能吗?
|
|