experiments/STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #2346 from mozilla/save-flags

Browse Source 
Save flag values next to checkpoints (Fixes #2345)
This commit is contained in:
Reuben Morais 2019-09-09 12:07:13 +02:00 committed by GitHub
commit 9c92f909b3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
DeepSpeech.py
View File

@ -471,6 +471,13 @@ def train():
best_dev_path = os.path.join(FLAGS.checkpoint_dir, 'best_dev') best_dev_path = os.path.join(FLAGS.checkpoint_dir, 'best_dev')
best_dev_filename = 'best_dev_checkpoint' 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())
initializer = tfv1.global_variables_initializer() initializer = tfv1.global_variables_initializer()
with tfv1.Session(config=Config.session_config) as session: with tfv1.Session(config=Config.session_config) as session: