Tensorflow数据读取器:重复读取相同的文件,直到培训结束

Tensorflow数据读取器:重复读取相同的文件,直到培训结束,tensorflow,Tensorflow,我已经使用教程在tensorflow中创建了一个模型,它基于我创建的一个示例数据集。模型将数据输入到输入文件中,直到它们结束,然后停止 以下是我读取文件的方式: filename_queue = tf.train.string_input_producer(filenames) reader = tf.FixedLengthRecordReader(record_bytes=record_bytes) result.key, value = reader.read(filename_queue)

我已经使用教程在tensorflow中创建了一个模型,它基于我创建的一个示例数据集。模型将数据输入到输入文件中,直到它们结束,然后停止

以下是我读取文件的方式:

filename_queue = tf.train.string_input_producer(filenames)
reader = tf.FixedLengthRecordReader(record_bytes=record_bytes)
result.key, value = reader.read(filename_queue)
这是session.run调用:

with tf.train.MonitoredTrainingSession(
                hooks=[tf.train.StopAtStepHook(last_step=max_steps)],  
                ) as mon_sess:

    while not mon_sess.should_stop():
        mon_sess.run(train_op)
文件结束时,内部循环停止

我希望模型继续运行,直到达到
最大步数
,根据需要重复读取相同的文件。


我知道这可以通过在
提要中向会话传递数据来手动完成,但我希望有一种更简单的方法允许我仍然使用内置的文件读取器。

字符串输入生成器的
num\u epochs
参数对您有用吗?也可能有用。
string\u input\u producer
num\u epochs
参数对您有用吗?也可能有用。