我有一些用tf.Session低级API编写的代码,由于我想在TPU上使用它,因此最好将它重写为tf.Estimator API,因为有一个用于TPU加速的TPUEstimator类。
是否有执行此操作的标准方法或有关如何执行此操作的指南?
谢谢你的帮助!
最佳答案
一个很好的初步指南可以在这里找到:
https://cloud.google.com/tpu/docs/using-estimator-api
以及此页面上“ TPU估算器”部分中的更多详细信息:
https://www.tensorflow.org/guide/using_tpu
如果您在阅读了这些文档以及它们链接到的其他文档之后还有其他疑问,请继续询问您更详细的问题,并使用google-cloud-tpu标签对其进行标记。