0
我试图将摘要添加到我的异步运行的TensorFlow图。我已经在单线程的情况下运行了所有的东西,但是一旦我进入多线程,总结似乎就消失了。这里有一个玩具的例子,我试图做TensorFlow与线程总结
import tensorflow as tf # 1.1.0
import threading
class Worker:
def __init__(self):
self.x = tf.Variable([1, -2, 3], tf.float32, name='x')
self.y = tf.Variable([-1, 2, -3], tf.float32, name='y')
self.dot_product = tf.reduce_sum(tf.multiply(self.x, self.y))
tf.summary.scalar("Dot_Product", self.dot_product)
def work(self):
for i in range(10):
SESS.run(self.dot_product)
# Write summary
summary_str = SESS.run(tf.summary.merge_all())
WRITER.add_summary(summary_str, i)
WRITER.flush()
COORD = tf.train.Coordinator()
SESS = tf.Session()
WRITER = tf.summary.FileWriter(SUMMARY_DIR, SESS.graph)
# Single Thread case
w = Worker()
SESS.run(tf.global_variables_initializer())
print(tf.get_collection(tf.GraphKeys.SUMMARIES))
w.work()
这工作正常。但是,如果我去的多线程:
# Multi-thread case
workers = [Worker() for i in range(4)]
SESS.run(tf.global_variables_initializer())
print(tf.get_collection(tf.GraphKeys.SUMMARIES))
worker_threads = []
for worker in workers:
job = lambda: worker.work()
t = threading.Thread(target=job)
t.start()
worker_threads.append(t)
COORD.join(worker_threads)
每当tf.summary.merge_all()
叫我得到这样的错误是由于事实,即它不能看到任何摘要:
Exception in thread Thread-2:
Traceback (most recent call last):
File "/usr/lib/python3.5/threading.py", line 914, in _bootstrap_inner
self.run()
File "/usr/lib/python3.5/threading.py", line 862, in run
self._target(*self._args, **self._kwargs)
File "/home/anjum/PycharmProjects/junk.py", line 43, in <lambda>
job = lambda: worker.work()
File "/home/anjum/PycharmProjects/junk.py", line 22, in work
summary_str = SESS.run(tf.summary.merge_all())
File "/usr/local/lib/python3.5/dist-
packages/tensorflow/python/client/session.py", line 778, in run
run_metadata_ptr)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/client/session.py", line 969, in _run
fetch_handler = _FetchHandler(self._graph, fetches, feed_dict_string)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/client/session.py", line 408, in __init__
self._fetch_mapper = _FetchMapper.for_fetch(fetches)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/client/session.py", line 227, in for_fetch
(fetch, type(fetch)))
TypeError: Fetch argument None has invalid type <class 'NoneType'>
如果我把里面print(tf.get_collection(tf.GraphKeys.SUMMARIES))
work()
,返回一个空列表。所以这意味着我的总结在某处丢失了。
有人请解释如何正确使用多线程摘要?