Estimator

时间:2017-05-12 16:16:05

标签: python tensorflow

我试图让我的模型使用tf.contrib.learn.Estimator和自定义模型函数来执行分布式培训。

Tensorflow documentation指定应使用tf.device函数手动将作业(ps和/或worker)放置在模型中。但是,本文档引用了声明模型的旧方法,而不使用Estimator类和模型函数。

我发现一些information声明新的Estimator"处理模型训练和评估的大部分复杂的分布式逻辑"。这是否意味着仍然需要在模型中手动放置各种worker / ps作业?如果是,那么应该从Tensorflow文档中的代码声明它们的方式是否有任何变化(即使用tf.device)?

1 个答案:

答案 0 :(得分:0)

在挖掘了一下之后,我发现了this code sample并且能够使分布式TF与Estimator一起工作,而无需在模型函数中手动放置各种tf.device