Daemonize TensorFlow child processes.

Required in order to let the training code exit cleanly when finished.
ssapo · Apr 3, 2018 · 00a4357 · 00a4357
1 parent 7f8ff14
commit 00a4357
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/training/tf/chunkparser.py b/training/tf/chunkparser.py
@@ -119,7 +119,8 @@ def __init__(self, chunkdatasrc, shuffle_size=1, sample=1, buffer_size=1, batch_
         for _ in range(workers):
             read, write = mp.Pipe(duplex=False)
             mp.Process(target=self.task,
-                    args=(chunkdatasrc, write)).start()
+                       args=(chunkdatasrc, write),
+                       daemon=True).start()
             self.readers.append(read)
             write.close()
         self.init_structs()