TensorFlow 的目标剪枝找不到节点

Question

我使用 TensorFlow API 编写了一个 Python 脚本，其中包括一个 SummaryWriter转储图形定义，以便我可以在 TensorBoard 中查看它。

运行脚本时，出现 NotFoundError被抛出说PruneForTargets: Some target nodes not found: Reading/data_queue_EnqueueMany_1 .顾名思义，有问题的节点是由 enqueue_many 创建的。调用 FIFOQueue (然后在 QueueRunner 中启动)；它确实存在，并且可以在 TensorBoard 中清楚地看到。

什么可能导致 TensorFlow 找不到某些节点？

Answer 1

这是一个已知问题，当您在向图中添加更多节点之前启动访问 TensorFlow 图(例如您的 QueueRunner)的线程时会发生此问题。 (底层 tf.Graph 数据结构对于并发读写不是线程安全的。)

解决办法是搬家 tf.train.start_queue_runners(sess) (以及任何其他启动线程的代码)在构建最后一个节点之后。仔细检查这一点的一种方法是添加对 tf.get_default_graph().finalize() 的调用。立即调用 start_queue_runners() .