Как конвертировать модель tf.Keras в TPU, используя TensorFlow 2.0 в Google Colab?

Question

Поскольку TF 2.0 не имеет слоя tf.contrib, как мне преобразовать мою модель для запуска обучения на TPU, не имея доступа к tf.contrib.tpu.keras_to_tpu_model()?

Я пытался найти код, но все он работает на TensorFlow 1.x

Мои данные в .npy, у меня есть простая модель, и я использую только model.compile() и model.fit() для обучения, но похоже, что модель работает на CPU (занимает 30 минут / эпоха против 2 минут / эпох на GPU).

Answer 1

Гуглер-Вольф подтвердил, что мы пока не можем использовать TF 2.0 в Colab с TPU (сообщается 15 апреля 2019 года):

TPU, который вы получите через Colab, работает под управлением TF 1.x. Когда вы устанавливаете ночной 2.0 пипс на свою виртуальную машину Jupyter, это не меняет TPU. В результате вы получаете несоответствие между тем, что работает на вашем экземпляре Jupyter, и тем, что имеет TPU.

И согласно https://github.com/tensorflow/tensorflow/issues/24412, поддержка TPU для TensorFlow 2.0 еще не завершена.

Решением будет отслеживание вышеуказанной проблемы и ожидание выхода TF 2.0.