将tf.Session重写为tf.Estimator API

时间:2018-08-29 16:34:06

标签: python tensorflow google-cloud-tpu

我有一些用tf.Session低级API编写的代码,由于我想在TPU上使用它,因此最好将其重写为tf.Estimator API,因为有一个用于TPU加速的TPUEstimator类。

是否有执行此操作的标准方法或有关如何执行此操作的指南?

感谢您的帮助!

1 个答案:

答案 0 :(得分:2)

可以在这里找到很好的初始指南:

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

以及此页面上“ TPU估算器”部分中的更多详细信息:

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

如果在阅读这些文档及其链接到的其他文档后还有其他疑问,请继续询问您更详细的问题,并使用google-cloud-tpu标签对其进行标记。