Fixed #546
This commit is contained in:
parent
958db24232
commit
843f55dbe2
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue