keras-onnx icon indicating copy to clipboard operation
keras-onnx copied to clipboard

keras2onnx: LSTM bidirectional layer error

Open chinmaychinara91 opened this issue 4 years ago • 0 comments

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 ?

chinmaychinara91 avatar Apr 06 '21 21:04 chinmaychinara91