TFX管道是否支持tpu上的培训?

Question

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

任何关于这将是非常感激的反馈!谢谢你！

注意:我使用TF v2

Answer 1

是的，TFX Pipeline为启用TPU的Estimators提供开箱即用的TPU支持，相同的model_config.model可以用于配置estimator，无论它是在CPU上训练还是在TPU上训练。

要进行TPU培训，您应该遵循以下操作:

1. 配置 tpu_footprint 在您的部署文件和所需的
2. 包含空 TPUSettings
3. 作为可选操作，中配置特定于tpuestimator的参数 model_config.tpu_estimator 。