2

DNNClassifierGoogle 的 TPU 要求您将 tensorflow Estimators 移植到TPUEstimators ，但我似乎无法弄清楚“罐装”估计器（如使用，tf.data.Dataset但我可能是错的。任何帮助表示赞赏。

4

1 回答 1

3

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

需要改变的东西...

  • 确保优化器包含在 tf.contrib.tpu.CrossShardOptimizer
  • tf.summary ops 需要删除或替换为 noops。
  • 让 Estimator 的 model_fn 返回 TPUEstimatorSpec
  • 强制执行诸如静态批量大小之类的事情（我相信 XLA 需要）

可能还有一些其他的东西。

于 2018-08-08T00:58:57.527 回答