TFX管道是否支持TPU上的培训？

Question

我是TFX的新手，我想知道是否有可能创建一个TFX管道，该管道可以使用来自TPUStrategy的TPUClusterResolver在TPU节点上训练Keras模型。看看TFX文档，我不清楚这是否可能。

任何有关这方面的反馈将是非常感谢的！谢谢!

注:我使用TF v2

Answer 1

是的，TFX管道支持TPU开箱即用的TPU启用估计器，无论它是在CPU上训练还是在TPU上训练，都可以使用相同的model_config.model来配置估计器。

要在TPU接受培训，你应该采取以下行动：

在您的部署文件中配置一个tpu_footprint，在您的服务文件中配置所需的训练器节。

• 在您的custom_config文件.

• 中包含一个空的消息，并且作为一个可选操作，在model_config.tpu_estimator.中配置一个TPUEstimator特定的args。