From 66a39088180052813c80a33919b85ff976b6e076 Mon Sep 17 00:00:00 2001 From: Yibing Liu Date: Fri, 12 Jan 2018 23:33:03 +0800 Subject: [PATCH] Adjust the order of scorer init & probs infer in infer.py --- infer.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/infer.py b/infer.py index 1539fbaa..5dd9b406 100644 --- a/infer.py +++ b/infer.py @@ -90,17 +90,18 @@ def infer(): # decoders only accept string encoded in utf-8 vocab_list = [chars.encode("utf-8") for chars in data_generator.vocab_list] + if args.decoding_method == "ctc_beam_search": + ds2_model.init_ext_scorer(args.alpha, args.beta, args.lang_model_path, + vocab_list) + + ds2_model.logger.info("start inference ...") probs_split = ds2_model.infer_probs_batch(infer_data=infer_data, feeding_dict=data_generator.feeding) if args.decoding_method == "ctc_greedy": - ds2_model.logger.info("start inference ...") result_transcripts = ds2_model.infer_batch_greedy( probs_split=probs_split, vocab_list=vocab_list) else: - ds2_model.init_ext_scorer(args.alpha, args.beta, args.lang_model_path, - vocab_list) - ds2_model.logger.info("start inference ...") result_transcripts = ds2_model.infer_batch_beam_search( probs_split=probs_split, beam_alpha=args.alpha, -- GitLab