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