为了监控我的模型在我的评估数据集上的性能,我使用tf.train.string_input_producer
作为.tfr文件上的文件名队列,然后我将解析后的示例提供给tf.train.batch
函数,该函数生成批量的固定的大小。
假设我的评估数据集包含761个示例(素数)。要准确读取所有示例一次,我必须有一个除以761的批量大小,但没有这样的,除了1将太慢和761不适合我的GPU。是否有任何标准方法可以准确读取每个示例一次?
实际上,我的数据集大小不是761,但是在50-300的合理范围内没有数字将它精确划分。此外,我正在使用许多不同的数据集,并且找到一个大致划分每个数据集中的示例数量的数字可能很麻烦。
请注意,使用num_epochs
tf.train.string_input_producer
参数无法解决问题。
谢谢!