Hi, I want to use TensorRT to optimize my inference model which is a CRNN architecture trained in keras, I can generate the tensorflow.pb file using the keras model, but when I try to load the tensorflow.pb file to generate tensorrt model, the error as below was happened:
ValueError: Input 0 of node bi_rnn1_1/while/ReadVariableOp/Enter was passed float from bi_rnn1_1/forward_gru_1/kernel:0 incompatible with expected resource.
I have reference some topic :
but it not work
and I think there is something wrong with rnn because I can generate a tensorRT engine using CNN model as fine.
Do you have any idea to solve these question?
the detail of code was attached in the below