Dear all members,
Recently, I have been working on converting LSTM models into HEF models.
As shown in the attached image, I attempted the conversion but encountered some unknown errors, as listed below.
I have already checked that the ONNX file works fine, but I couldn’t find any other way to resolve these issues.
If anyone with expertise in converting LSTM models can provide some hints, I would greatly appreciate it.
Thank you for reading.
Have a great day.
Jusung Kang
** Convert line **
runner = ClientRunner(hw_arch=chosen_hw_arch)
hn, npz = runner.translate_onnx_model(
onnx_path,
# start_node_names=[“input”],
start_node_names=[‘input’, ‘h0’, ‘c0’],
# end_node_names=[“output”],
end_node_names=[‘/linear/Gemm’, ‘hn’, ‘cn’],
# net_input_shapes={“input”: [batch_size, 1, 14]},
net_input_shapes={“input”: [batch_size, 30, 14], “h0”: [1, batch_size, 64], “c0”: [1, batch_size, 64]},
)
** Taken Errors**
[info] Extracted ONNXRuntime meta-data for Hailo model (completion time: 00:00:00.02)
[info] Simplified ONNX model for a parsing retry attempt (completion time: 00:00:00.03)
Traceback (most recent call last):
File “/home/jskang/venv_JSK/GRPC_JSK_v2_RT/lib/python3.9/site-packages/hailo_sdk_client/sdk_backend/parser/parser.py”, line 235, in translate_onnx_model
parsing_results = self._parse_onnx_model_to_hn(
File “/home/jskang/venv_JSK/GRPC_JSK_v2_RT/lib/python3.9/site-packages/hailo_sdk_client/sdk_backend/parser/parser.py”, line 316, in _parse_onnx_model_to_hn
return self.parse_model_to_hn(
File “/home/jskang/venv_JSK/GRPC_JSK_v2_RT/lib/python3.9/site-packages/hailo_sdk_client/sdk_backend/parser/parser.py”, line 367, in parse_model_to_hn
fuser = HailoNNFuser(converter.convert_model(), net_name, converter.end_node_names)
File “/home/jskang/venv_JSK/GRPC_JSK_v2_RT/lib/python3.9/site-packages/hailo_sdk_client/model_translator/translator.py”, line 83, in convert_model
self._create_layers()
File “/home/jskang/venv_JSK/GRPC_JSK_v2_RT/lib/python3.9/site-packages/hailo_sdk_client/model_translator/edge_nn_translator.py”, line 40, in _create_layers
self._add_direct_layers()
File “/home/jskang/venv_JSK/GRPC_JSK_v2_RT/lib/python3.9/site-packages/hailo_sdk_client/model_translator/edge_nn_translator.py”, line 122, in _add_direct_layers
self._layer_callback_from_vertex(vertex)
File “/home/jskang/venv_JSK/GRPC_JSK_v2_RT/lib/python3.9/site-packages/hailo_sdk_client/model_translator/onnx_translator/onnx_translator.py”, line 460, in _layer_callback_from_vertex
consumed_vertices = self._create_lstm_layer(vertex)
File “/home/jskang/venv_JSK/GRPC_JSK_v2_RT/lib/python3.9/site-packages/hailo_sdk_client/model_translator/onnx_translator/onnx_translator.py”, line 2041, in _create_lstm_layer
forward_params, backward_params, direction, consumed_vertices = vertex.get_lstm_info()
File “/home/jskang/venv_JSK/GRPC_JSK_v2_RT/lib/python3.9/site-packages/hailo_sdk_client/model_translator/onnx_translator/onnx_graph.py”, line 1330, in get_lstm_info
info_dict[param] = const.parse_raw_data()
File “/home/jskang/venv_JSK/GRPC_JSK_v2_RT/lib/python3.9/site-packages/hailo_sdk_client/model_translator/onnx_translator/onnx_graph.py”, line 620, in parse_raw_data
parsed_data = numpy_helper.to_array(self._info.attribute[0].t)
AttributeError: attribute