Fixed #546

2017-04-20 12:46:51 +02:00 · 2017-04-20 12:46:51 +02:00 · 843f55dbe2
parent 958db24232
commit 843f55dbe2
1 changed files with 1 additions and 0 deletions
--- a/util/data_set_helpers.py
+++ b/util/data_set_helpers.py
@ -66,6 +66,7 @@ class DataSet(object):
        self._coord = coord
        batch_threads = [Thread(target=self._populate_batch_queue, args=(session,)) for i in range(self._thread_count)]
        for batch_thread in batch_threads:
+            self._coord.register_thread(batch_thread)
            batch_thread.daemon = True
            batch_thread.start()
        return batch_threads