Как говорит ошибка, физические ограничения вашего оборудования были достигнуты. Требуется больше памяти GPU, чем доступно.
Вы можете предотвратить это, уменьшив размер пакета моделей или изменив разрешение входной видеопоследовательности.
В качестве альтернативы, вы можете попробовать использовать облачное обучение Google для получения дополнительных аппаратных ресурсов, однако оно не является бесплатным.
https://cloud.google.com/tpu/