从 CSV 文件训练 - 每个时期使用每个训练示例答案

【问题标题】：Training from CSV file - use every train example per epoch从 CSV 文件训练 - 每个时期使用每个训练示例
【发布时间】：2015-12-16 14:47:01
【问题描述】：

我有一个包含 200.000 个训练样本的 CSV 文件，我想用它来训练我的网络。

我正在使用InputProducer 和DecodeCSV 来获取数据。然后我通过shuffle_batch 运行所有数据，其中我设置了batch_size=50、min_after_dequeue=10000 和capacity=min_after_dequeue + 3 * batch_size。

然后我运行一个循环并反复调用sess.run()。

我的问题是我现在想运行几个 epoch。在每个时期，我都想消耗整个训练集。我不认为当前的设置可以做到这一点。我该怎么做呢？我什至不确定，我是否完全了解shuffle_batch 的内部工作原理及其参数。

提前谢谢你。

【问题讨论】：

标签： tensorflow

【解决方案1】：

队列应该在 epoch 结束时阻塞。当这种情况发生时，你会知道你已经用尽了训练集。此相关问题中的更多信息：Tensor Flow shuffle_batch() blocks at end of epoch

【讨论】：

所以如果coord.should_stop() 为真，我的输入队列已经清空。如何告诉我的 input_producer 再次填满队列？这样做的推荐方法是什么？