我试图让我的模型使用tf.contrib.learn.Estimator
和自定义模型函数来执行分布式培训。
Tensorflow documentation指定应使用tf.device
函数手动将作业(ps和/或worker)放置在模型中。但是,本文档引用了声明模型的旧方法,而不使用Estimator
类和模型函数。
我发现一些information声明新的Estimator
"处理模型训练和评估的大部分复杂的分布式逻辑"。这是否意味着仍然需要在模型中手动放置各种worker / ps作业?如果是,那么应该从Tensorflow文档中的代码声明它们的方式是否有任何变化(即使用tf.device
)?