Merge pull request #2869 from lissyx/scorer-fail-early
Add some early checks, for Scorer at first
This commit is contained in:
commit
7b8b678310
|
@ -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()
|
||||||
|
|
Loading…
Reference in New Issue