如何使用队列在tensorflow上多次浏览相同的数据集

时间:2017-04-24 03:46:37

标签: tensorflow

我有一个数据集。为了训练它,我想多次通过优化器。如何使用队列机制执行此操作?有没有这样做的标准方法?

1 个答案:

答案 0 :(得分:1)

基于队列的TensorFlow输入管道通常以某种"输入生成器"开头。阶段,例如tf.train.string_input_producer()tf.train.slice_input_producer()或通用tf.train.input_producer()。这些输入生成器每个都采用一个可选的num_epochs参数,并由一个"队列运行器"将输入num_epochs次重复排入输入管道的线程。

num_epochs=n传递给其中一个输入生成器函数将导致输入管道接收输入数据的n个副本。您也可以通过num_epochs=None无限重复输入。