onnx / keras-onnx

Convert tf.keras/Keras models to ONNX
Apache License 2.0
379 stars 109 forks source link

keras2onnx: LSTM bidirectional layer error #702

Open chinmaychinara91 opened 3 years ago

chinmaychinara91 commented 3 years ago

Hi, I have a bidirectional LSTM trained keras model (.h5) which is as under: bidirectional_LSTM.zip

I use a simple script for conversion:

from tensorflow.python.keras.models import load_model
import onnx
import keras2onnx
onnx_model_name = 'bidirectional_LSTM.onnx'

model = load_model('bidirectional_LSTM.h5')
onnx_model = keras2onnx.convert_keras(model, model.name)
onnx.save_model(onnx_model, onnx_model_name)

I get the error: ValueError: Unsupported class for Bidirectional layer: <class 'tensorflow.python.keras.layers.recurrent_v2.LSTM'>

Using tensorflow_gpu version 2.3.1

How do I fix this ?