设置string_input_producer时Tensorflow num_epochs行为不当

Question

在设置string_input_producer时，num_epochs似乎并不是队列字符串。

使用下面的代码，程序打印，这是不正确的。

`import tensorflow as tf
sess = tf.InteractiveSession()
filenames = ["1", "2", "3"]
filename_queue = tf.train.string_input_producer(filenames, num_epochs=10)
test_value = tf.convert_to_tensor(filename_queue.size())

coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(sess=sess, coord=coord)

print(sess.run([test_value]))`

但如果我干掉num_epochs，

`import tensorflow as tf
sess = tf.InteractiveSession()
filenames = ["1", "2", "3"]
filename_queue = tf.train.string_input_producer(filenames)
test_value = tf.convert_to_tensor(filename_queue.size())

coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(sess=sess, coord=coord)

print(sess.run([test_value]))`

它按预期打印3。

还有其他人遇到同样的问题吗？

Answer 1

解决方案是添加以下一行：

sess.run(tf.initialize_all_variables())

...before启动队列运行程序。tf.train.string_input_producer()函数在内部创建一个TensorFlow变量来跟踪当前的时代索引，并且它必须在第一次使用之前被初始化(当您启动队列运行程序时)。

Answer 2

现在，使用

sess.run(tf.local_variables_initializer())

在启动队列运行程序之前。因为如果生产者()不是None，则num_epochs将创建本地计数器。