commit
70da866e03
@ -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…
x
Reference in New Issue
Block a user