GCP TPU上的固定模型

Question

Google的TPU要求您将张量流估计器移植到TPUEstimators，但我似乎无法弄清楚“固定”估计器（例如DNNClassifier）需要进行哪些更改-它似乎只有输入函数确实需要更改才能使用tf.data.Dataset，但我可能是错的。任何帮助表示赞赏。

Answer 1

TPU罐头估算器正在开发中。但是，为了回答您的问题，我相信您只需要...

需要改变的事物...

• 确保优化器包装在tf.contrib.tpu.CrossShardOptimizer中
• tf.summary操作需要删除或用noop代替。
• 具有Estimator的model_fn返回TPUEstimatorSpec
• 强制执行诸如静态批处理大小之类的事情（我相信XLA要求这样做）

可能还有其他事情。