准确地说出每个例子一次

时间:2016-09-20 16:27:14

标签: tensorflow

为了监控我的模型在我的评估数据集上的性能,我使用tf.train.string_input_producer作为.tfr文件上的文件名队列,然后我将解析后的示例提供给tf.train.batch函数,该函数生成批量的固定的大小。

假设我的评估数据集包含761个示例(素数)。要准确读取所有示例一次,我必须有一个除以761的批量大小,但没有这样的,除了1将太慢和761不适合我的GPU。是否有任何标准方法可以准确读取每个示例一次?

实际上,我的数据集大小不是761,但是在50-300的合理范围内没有数字将它精确划分。此外,我正在使用许多不同的数据集,并且找到一个大致划分每个数据集中的示例数量的数字可能很麻烦。

请注意,使用num_epochs tf.train.string_input_producer参数无法解决问题。

谢谢!

1 个答案:

答案 0 :(得分:1)

您可以在this example中使用reader.read_up_to。您的最后一批产品将更小,因此您需要确保您的网络不会在任何地方硬连接批量大小