Tensorflow数据读取器:重复读取相同的文件,直到培训结束
我已经使用教程在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)
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
参数对您有用吗?也可能有用。