Add some early checks, for Scorer at first

Fixes #2807
This commit is contained in:
Alexandre Lissy 2020-03-31 15:11:55 +02:00
parent 83d22e591b
commit 950d097ca1
1 changed files with 9 additions and 0 deletions

View File

@ -896,8 +896,17 @@ def do_single_file_inference(input_file_path):
print(decoded[0][1]) print(decoded[0][1])
def early_checks():
# Check for proper scorer early
if FLAGS.scorer_path:
scorer = Scorer(FLAGS.lm_alpha, FLAGS.lm_beta,
FLAGS.scorer_path, Config.alphabet)
del scorer
def main(_): def main(_):
initialize_globals() initialize_globals()
early_checks()
if FLAGS.train_files: if FLAGS.train_files:
tfv1.reset_default_graph() tfv1.reset_default_graph()