From ebf5b79fae9d8697df18dea970d30c495418f62a Mon Sep 17 00:00:00 2001 From: vincentg Date: Fri, 15 Jun 2018 15:47:07 -0700 Subject: [PATCH] some hack to make windows save the model when do ctrl+c --- python/unitytrainers/trainer_controller.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/unitytrainers/trainer_controller.py b/python/unitytrainers/trainer_controller.py index b0116651b8..27905cd91b 100644 --- a/python/unitytrainers/trainer_controller.py +++ b/python/unitytrainers/trainer_controller.py @@ -276,6 +276,7 @@ def start_learning(self): if global_step != 0 and self.train_model: self._save_model(sess, steps=global_step, saver=saver) except KeyboardInterrupt: + print('--------------------------Now saving model-------------------------') if self.train_model: self.logger.info("Learning was interrupted. Please wait while the graph is generated.") self._save_model(sess, steps=global_step, saver=saver)