具有协调器的估计器作为用于在张量流中以分布式方式读取输入数据的输入功能

Question

CNN cifar10教程（张量流教程）给出了一个低级API用于读取数据的示例，作为一个独立的工作来训练模型（具有多个GPU）。 是否可以使用具有低级线程支持和多/单GPU培训的高级API Estimator？

我正在寻找一种结合两者的方法：

• 来自高级API的自定义估算工具，详情为https://www.tensorflow.org/extend/estimators

• input_fn作为一个队列，它提供了https://www.tensorflow.org/programmers_guide/reading_data中为协调员类描述的相同功能

  coord = tf.train.Coordinator()
  threads = tf.train.start_queue_runners(sess=sess, coord=coord)
  

这对我来说并不简单！

Answer 1

我将代码推送到here 使用estimator时，它支持input_fn作为队列。高级API估算器，具有低级线程支持和多/单GPU培训。并且您可以轻松获得所有需要的客户代码。