Google 的 TPU 要求您将 tensorflow 估算器移植到 TPUEstimators,但我似乎无法弄清楚“ jar 装”估算器(例如 DNNClassifier
)需要进行什么样的更改) - 似乎只有输入函数确实需要更改才能使用 tf.data.Dataset 但我可能是错的。任何帮助表示赞赏。
最佳答案
TPU 固定估算器正在开发中。但是,要回答你的问题,我相信你只需要......
需要改变的事情......
- 确保优化器包含在 tf.contrib.tpu.CrossShardOptimizer 中
- tf.summary 操作需要删除或替换为 noop。
- 让 Estimator 的 model_fn 返回 TPUEstimatorSpec
- 强制执行诸如静态批量大小之类的操作(我相信 XLA 需要这样做)
可能还有其他一些事情。
关于tensorflow - GCP TPU 上的封装模型,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/51227912/