Перепишите tf.Session в tf.Estimator API.

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

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

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


person craft    schedule 29.08.2018    source источник


Ответы (1)


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

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

и более подробно в разделе «Оценщик TPU» на этой странице:

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

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

person Derek T. Jones    schedule 31.08.2018