python - tf.train.range_input_producer(epoch_size, shuffle=True) 不会终止也不会引起 CPU/GPU 负载

Question

我在处理 RNN 时遇到了一个奇怪的问题。我正在关注 TensorFlow RNN Tutorial 并尝试我自己的(更简单的)实现，它受到 R2RT's Blog Post: Recurrent Neural Networks in Tensorflow I 的启发。

调试后我确定问题出在 tensorflow.models.rnn.ptb.reader.py(第 115 行)中的 ranger_input_producer。

我在最小的示例中将其隔离:

import tensorflow as tf

epoch_size = 20
i = tf.train.range_input_producer(epoch_size, shuffle=False).dequeue()

这就是 ptb_producer(epoch_size 的变量值)。事实证明，这段代码并没有终止(我什至没有调用任何 session.run(...) 也没有使用 CPU。我猜队列正在等待 suggested by Daving Wong .

有什么线索吗？谢谢

轻声

Answer 1

如果你只是使用代码将 tf.Session() 作为 sess:,你必须明确地打开线程threads = tf.train.start_queue_runners()。但是在 ptb_word_lm.py 中，它使用这样的代码sv = tf.train.Supervisor() with sv.managed_session() as sess:,Supervisor() 函数包含一些隐式启动线程的东西