I’m trying to convert a TF model (based on LSTM layers) to a TRT engine, but having some troubles. (Actually I tried with different models and reaching the same problem, so I think it’s not problem of the model)
The model I’m currently using is:
text_lstm_model.onnx (3.1 MB)
I’m using trtexec to build the engine. The error I’m getting is:
----- Parsing of ONNX …/text_lstm_model.onnx is Done ----
[07/28/2021-09:47:01] [E] [TRT] Repeated layer name: while/MatMul_1 (layers must have distinct names)
[07/28/2021-09:47:01] [E] [TRT] Network validation failed.
[07/28/2021-09:47:01] [E] Engine creation failed
[07/28/2021-09:47:01] [E] Engine set up failed
It seems that the parsing is OK, but the problem is when building the network. My guess is that there is some problem with the while loop. What should I do? Is there any trick to implement LSTM based models?