From f854dcb83f2fc67ab58c577a0f6ab2ee10cf5970 Mon Sep 17 00:00:00 2001 From: "gw0 [http://gw.tnode.com/]" Date: Fri, 5 Feb 2016 15:36:14 +0100 Subject: [PATCH] Fix program termination when threads are used in fit_generator(). --- keras/models.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/keras/models.py b/keras/models.py index 46a60582c54..d3da5a3355f 100644 --- a/keras/models.py +++ b/keras/models.py @@ -995,6 +995,7 @@ def generator_task(): generator_threads = [threading.Thread(target=generator_task) for _ in range(nb_worker)] for thread in generator_threads: + thread.daemon = True thread.start() self.stop_training = False @@ -1452,6 +1453,7 @@ def generator_task(): generator_threads = [threading.Thread(target=generator_task) for _ in range(nb_worker)] for thread in generator_threads: + thread.daemon = True thread.start() self.stop_training = False