TensorFlow团队一直在努力开发一个新的功能(在撰写本文时仍处于开发状态),可让你训练tf.keras模型而无需先将其转换为Estimator,只需几行额外的代码!那个工作流程也很棒。下面我会重点介绍Estimators API。无论选择哪一个都取决于你!
我们需要记住的是,只要用tf.keras.estimator.model_to_estimator方法将tf.keras.Model转换为tf.estimator.Estimator对象,tf.keras.Model就可以使用tf.estimator API训练。转换后,我们可以应用Estimators提供的机制在不同的硬件配置训练(代码链接在文末)。
教程链接:tf.estimator示例