Error when converting Keras ConvLSTM2D layer into ONNX
YizhouFan opened this issue · 1 comments
YizhouFan commented
My code snippet for conversion:
MODEL_NAME = "data/pie/intention/context_loc_pretrained/model"
model = load_model(MODEL_NAME + ".h5")
onnx_model = keras2onnx.convert_keras(model, model.name, target_opset=14)
temp_model_file = MODEL_NAME + ".onnx"
onnx.save_model(onnx_model, temp_model_file)
Error message:
ValueError: Unable to find out a correct type for tensor type = 0 of conv_lst_m2d_1/while/TensorArrayWrite/TensorArrayWriteV3/Enter:0
Network definition:
See the open-source repository here,
https://github.com/aras62/PIEPredict/blob/df76b6512dd8afa8e695025e306a9a21f54b1179/pie_intent.py#L451
Versions:
Keras2onnx: 1.9.0
onnxconverter-common: 1.9.0
tensorflow: 1.14.0
Keras: 2.2.4
nirlviz commented
Happens to me too.
Any updates about this?