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

Save flag values next to checkpoints

Browse Source
This commit is contained in:
Reuben Morais 2019-09-06 11:35:10 +02:00
parent 935ede3f28
commit ba2d29b36f
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
DeepSpeech.py
View File

@ -471,6 +471,11 @@ 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
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: