added daemin for threads

wlads · Sep 13, 2016 · 514dd3f · 514dd3f
1 parent f0ba5ca
commit 514dd3f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tflearn/data_flow.py b/tflearn/data_flow.py
@@ -149,6 +149,7 @@ def start(self, reset_status=True):
                       for i in range(self.num_threads)]
         self.threads = bi_threads + fd_threads
         for t in self.threads:
+            t.daemon = True
             t.start()
 
     def stop(self):
@@ -161,7 +162,9 @@ def stop(self):
         for i in range(self.num_threads):
             self.batch_ids_queue.put(False)
         # Launch a Thread to wait for processing scripts to finish
-        threading.Thread(target=self.wait_for_threads).start()
+        t = threading.Thread(target=self.wait_for_threads)
+        t.daemon = True
+        t.start()
 
     def reset(self):
         """ reset.