Error when converting Keras ConvLSTM2D layer into ONNX - Githubissues

onnx / keras-onnx

Convert tf.keras/Keras models to ONNX

Apache License 2.0

379 stars 110 forks source link

Error when converting Keras ConvLSTM2D layer into ONNX #731

Open YizhouFan opened 3 years ago

YizhouFan commented 3 years ago

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 2 years ago

Happens to me too. Any updates about this?