Tensorflow:使用整个验证数据集来验证网络,以及如何使用Queue实现它

时间:2016-11-23 15:05:11

标签: tensorflow

我有一个包含10k个样本的验证数据集,存储在TFRecords文件中。我使用队列来读取文件作为网络的输入,并将batch_size设置为128.由于10k / 128 = 78.125,所以队列可以出列78次,但剩下的则为10k - 128 * 78 = 16)样本未经验证,因为在78次出队操作后,程序将抛出OutOfRangeError。

但是,我确实要验证确切的10k样本,包括最终的16个样本。如何在TensorFlow中使用队列进行此操作?

1 个答案:

答案 0 :(得分:0)

Neal的评论是一个很好的解决方案:首先,您可以将批量大小设置为数据集大小的精确倍数。或者,您可以运行最终的"尾随"示例使用feed_dict手动添加结果,并将其结果添加到您正在跟踪的内容中。