Create checkpoint dir before writing flags file in it

2019-09-06 13:55:55 +02:00 · 2019-09-06 13:55:55 +02:00 · 371e73eb69
commit 371e73eb69
parent ba2d29b36f
1 changed files with 2 additions and 0 deletions
--- a/DeepSpeech.py
+++ b/DeepSpeech.py
@ -472,6 +472,8 @@ def train():
    best_dev_filename = 'best_dev_checkpoint'

    # Save flags next to checkpoints
+    os.makedirs(FLAGS.checkpoint_dir, exist_ok=True)
+
    flags_file = os.path.join(FLAGS.checkpoint_dir, 'flags.txt')
    with open(flags_file, 'w') as fout:
        fout.write(FLAGS.flags_into_string())