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