Merge pull request mozilla#708 from mozilla/issue695

Fix mozilla#695; Checkpoint is not resuming from left off point
superkm · Jul 12, 2017 · 862815a · 862815a
2 parents d57652f + a0a9e9d
commit 862815a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/DeepSpeech.py b/DeepSpeech.py
@@ -1497,6 +1497,8 @@ def end(self, session):
         with tf.train.MonitoredTrainingSession(master='' if server is None else server.target,
                                                is_chief=is_chief,
                                                hooks=hooks,
+                                               checkpoint_dir=FLAGS.checkpoint_dir,
+                                               save_checkpoint_secs=FLAGS.checkpoint_secs,
                                                config=session_config) as session:
             try:
                 if is_chief: