'Ожидаемый аргумент `model` будет экземпляром` Model`, получил объект keras

Question

Я пытаюсь использовать ТПУ на колабе,

model.compile(optimizer='adam',
              loss='binary_crossentropy',
              metrics=[auc])

model = tf.contrib.tpu.keras_to_tpu_model(
        model,
        strategy=tf.contrib.tpu.TPUDistributionStrategy(

tf.contrib.cluster_resolver.TPUClusterResolver(TPU_ADDRESS)))

Я получил следующую ошибку.

Ожидается, что model аргумент будет Model экземпляром, получил <keras.engine.training.Model object at 0x7f8e828822e8>

Любая помощь приветствуется. Спасибо!

Answer 1

Возможно, попробуйте сначала преобразовать не скомпилированную модель в модель TPU, затем скомпилируйте модель TPU.

Как в этой статье, которую я нашел.https://medium.com/the-artificial-impostor/keras-for-tpus-on-google-colaboratory-free-7c00961fed69

А вот и ссылка на Google Colab https://colab.research.google.com/gist/ceshine/f196d6b030adb1ec3a8d0b50642709dc/keras-fashion-mnist-tpu.ipynb#scrollTo=pWEYmd_hIWg8