This commit is contained in:
Kelly Davis 2017-04-20 12:46:51 +02:00
parent 958db24232
commit 843f55dbe2
1 changed files with 1 additions and 0 deletions

View File

@ -66,6 +66,7 @@ class DataSet(object):
self._coord = coord self._coord = coord
batch_threads = [Thread(target=self._populate_batch_queue, args=(session,)) for i in range(self._thread_count)] batch_threads = [Thread(target=self._populate_batch_queue, args=(session,)) for i in range(self._thread_count)]
for batch_thread in batch_threads: for batch_thread in batch_threads:
self._coord.register_thread(batch_thread)
batch_thread.daemon = True batch_thread.daemon = True
batch_thread.start() batch_thread.start()
return batch_threads return batch_threads