Переписать tf.Session в API tf.Estimator

У меня есть некоторый код, который был написан с помощью низкоуровневого API tf.Session, и, поскольку я хочу использовать его на TPU, лучше переписать его в API tf.Estimator, поскольку для ускорения TPU существует класс TPUEstimator.

Есть ли стандартный способ сделать это или руководство о том, как подойти к этому?

Спасибо за любую помощь!

Источник

1 ответ

Хорошее начальное руководство можно найти здесь:

https://cloud.google.com/tpu/docs/using-estimator-api

и более подробно в разделе "Оценщик ТПУ" на этой странице:

https://www.tensorflow.org/guide/using_tpu

Если после прочтения этих и других документов, на которые они ссылаются, у вас возникнут дополнительные вопросы, ответьте на них, задав более подробные вопросы и пометив их тегом google-cloud-tpu.

Источник
Другие вопросы по тегам