GCP TPU 上的预制模型

Question

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

Answer 1

TPU罐头估计正在进行中。但是，为了回答你的问题，我相信你必须...

需要改变的东西...

• 确保在TF.Contrib.tpu.crossShardOpTimizer Li>中包装优化器
• TF.需要用诺沃删除或替换所有操作。
• 具有估算器的model_fn返回tpuestimatorspec
• 强制执行静态批量大小（哪个XLA所需的信息）

可能是其他一些东西。