tensorflow使用tf.train.string_input_producer

时间:2017-06-07 14:45:10

标签: python tensorflow

我使用tf.train.string_input_producer从tfRecord文件中读取数据。我想它会创建一个队列和管道,数据将自动加载并输入到我的模型中。但是,它停留在第一批,并显示此异常:

  

FailedPreconditionError(参见上面的回溯):尝试使用未初始化的值input_producer / limit_epochs / epochs

我的tfrecord是由tf.train.SequenceExample制作的,而不是tf.train.Example,在官方指南中没有明确的文档。

这里是重现我的问题的代码快照。 (我相信我的问题来自队列初始化或某事。因为似乎整个管道都挂断了)

xsp.persistence.tree.maxviews=5

1 个答案:

答案 0 :(得分:1)

尝试

init2 = tf.local_variables_initializer()
sess.run(init2)

tf.train.string_input_producer()中的Variabes(num_epochs或capacity)是局部变量。您必须使用局部变量初始值设定项对它们进行初始化,如上所示。

如果这有帮助,请告诉我。