Почему я получаю эту ошибку? valueerror: ожидается, что выходной тензор с индексом 0 будет иметь 3 измерения, найдено 2

Question

Мой проект основан на обнаружении объектов с помощью tflite на raspberry pi4, код отлично работает без кораллового ускорителя, однако, когда я пытаюсь ускорить обнаружение с помощью кораллового ускорителя Google, я получаю эту ошибку: (я запускаю код с помощью Google Collab)

valueerror: ожидается, что выходной тензор с индексом 0 будет иметь 3 измерения, найдено 2.

Это полный код с использованием совместной работы Google:

https://colab.research.google.com/github/khanhlvg/tflite_raspberry_pi/blob/main/object_detection/Train_custom_model_tutorial.ipynb

Пожалуйста, дайте мне подробное решение, спасибо :)

Я попытался изменить версию tflite на 2.5.0, но код не сработал.