python - 将 tf.Session 重写为 tf.Estimator API

Question

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

有没有标准的方法来做到这一点或如何处理这个问题的指南？

谢谢你的帮助！

Answer 1

可以在这里找到一个很好的初始指南：

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

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

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

如果您在阅读了这些文档和它们链接的其他文档后还有其他问题，请通过提出更详细的问题并使用 google-cloud-tpu 标签标记它们来跟进。