Я пытаюсь применить TensorRT к своему проекту, который имеет два уровня: обнаружение объектов (YOLOv3) и отслеживание объектов (модель LSTM). Обе модели прекрасно работают без TesnorRT. Однако я хочу развернуть свой стек на устройстве Jetson, что потребовало от меня использования TesnorRT для увеличения скорости и снижения энергопотребления. Мне удалось применить TensorRT на YOLOv3, однако мне не удалось заставить его работать с моделью LSTM.
Я получаю ошибку, что состояния не могут преобразовывать операции, хотя я уверен, что я преобразование тензоров (как я проверил вручную для всех преобразованных элементов и заморозил график).