fix for RuntimeError

r9y9 · Mar 3, 2018 · 5aea7cd · 5aea7cd
1 parent 4b66973
commit 5aea7cd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/train.py b/train.py
@@ -959,6 +959,9 @@ def restore_parts(path, model):
               clip_thresh=hparams.clip_thresh,
               train_seq2seq=train_seq2seq, train_postnet=train_postnet)
     except KeyboardInterrupt:
+        print("Interrupted!")
+        pass
+    finally:
         save_checkpoint(
             model, optimizer, global_step, checkpoint_dir, global_epoch,
             train_seq2seq, train_postnet)