keras-onnx
keras-onnx copied to clipboard
keras2onnx: LSTM bidirectional layer error
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 ?