Skip to content

Commit ebf5b79

Browse files
committed
some hack to make windows save the model when do ctrl+c
1 parent 060a9ef commit ebf5b79

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

python/unitytrainers/trainer_controller.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -276,6 +276,7 @@ def start_learning(self):
276276
if global_step != 0 and self.train_model:
277277
self._save_model(sess, steps=global_step, saver=saver)
278278
except KeyboardInterrupt:
279+
print('--------------------------Now saving model-------------------------')
279280
if self.train_model:
280281
self.logger.info("Learning was interrupted. Please wait while the graph is generated.")
281282
self._save_model(sess, steps=global_step, saver=saver)

0 commit comments

Comments
 (0)