From cf2d7e636a9fa6a1b08e5b5b437dc80fc7d818dc Mon Sep 17 00:00:00 2001 From: Catalin Voss Date: Mon, 2 Nov 2020 21:09:35 -0800 Subject: [PATCH] Make variables consistent --- training/deepspeech_training/train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/training/deepspeech_training/train.py b/training/deepspeech_training/train.py index 12df5dec..8bf7a354 100644 --- a/training/deepspeech_training/train.py +++ b/training/deepspeech_training/train.py @@ -900,21 +900,21 @@ def do_single_file_inference(input_file_path): features = create_overlapping_windows(features).eval(session=session) features_len = features_len.eval(session=session) - logits = outputs['outputs'].eval(feed_dict={ + probs = outputs['outputs'].eval(feed_dict={ inputs['input']: features, inputs['input_lengths']: features_len, inputs['previous_state_c']: previous_state_c, inputs['previous_state_h']: previous_state_h, }, session=session) - logits = np.squeeze(logits) + probs = np.squeeze(probs) if FLAGS.scorer_path: scorer = Scorer(FLAGS.lm_alpha, FLAGS.lm_beta, FLAGS.scorer_path, Config.alphabet) else: scorer = None - decoded = ctc_beam_search_decoder(logits, Config.alphabet, FLAGS.beam_width, + decoded = ctc_beam_search_decoder(probs, Config.alphabet, FLAGS.beam_width, scorer=scorer, cutoff_prob=FLAGS.cutoff_prob, cutoff_top_n=FLAGS.cutoff_top_n) # Print highest probability result