Я не могу загрузить обученную модель в Google Colab TPU – ошибка, вызванная пакетной формой ⇐ Python

[Anonymous]

Я использую блокноты в Kaggle и Google Colab. После того как я обучил свою модель в Kaggle (GPU P100) и попытался загрузить ее в Colab (TPU), я получил следующую ошибку:

Код: Выделить всё

TypeError: Error when deserializing class 'InputLayer' using config={'batch_shape': [None, 191], 'dtype': 'float32', 'sparse': False, 'name': 'input_layer'}.

Exception encountered: Unrecognized keyword arguments: ['batch_shape']

Раньше я загружал модели в графический процессор Colab T4 (а также в процессор), и все работало отлично. Кто-нибудь знает, что мне нужно сделать, чтобы перенести существующую модель на Colab TPU?
Я пытался выяснить, какие версии keras используются в каждой среде, поэтому я запустил это:< /p>

Код: Выделить всё

import tensorflow as tf
from tensorflow import keras

print(tf.__version__)
print(keras.__version__)

В Kaggle GPU P100 и Colab CPU это приводит к следующему:

Код: Выделить всё

2.15.0
3.4.1

В то время как в Google Colab TPU с указанной выше версией кода tensorflow отображается 2.15.0, а часть keras возвращает ошибку:

Код: Выделить всё

AttributeError: module 'keras.api._v2.keras' has no attribute '__version__'

Примечание:
Я не понимаю, почему Colab в настоящее время не позволяет мне использовать графический процессор. Я не превышал лимиты, так как не тренировался с графическим процессором последние 3 дня. По моему опыту, и графический процессор, и TPU имеют одинаковый предел, поэтому я удивлен, что TPU работает (ну... по крайней мере, он активируется), а графический процессор - нет.
В любом случае, может кто-нибудь помочь? с портированием моделей в ТПУ?

Подробнее здесь: https://stackoverflow.com/questions/789 ... atch-shape