diff --git a/deep_speech_2/decoder.py b/deep_speech_2/decoder.py index fc746c705695833dd53c7fb8c4f9d263b4ba2852..96e9118112a0fae6e51b441a99bdc2fa3fb792b4 100755 --- a/deep_speech_2/decoder.py +++ b/deep_speech_2/decoder.py @@ -199,7 +199,7 @@ def ctc_beam_search_decoder_nproc(probs_split, ''' Beam search decoder using multiple processes. - :param probs_seq: 3-D list with length num_time_steps, each element + :param probs_seq: 3-D list with length batch_size, each element is a 2-D list of probabilities can be used by ctc_beam_search_decoder. diff --git a/deep_speech_2/infer.py b/deep_speech_2/infer.py index 377aeb73cf9dd81eea05fceab0ef66e6420eac8c..0be89e617aae92691220e9981055867fad522d9b 100644 --- a/deep_speech_2/infer.py +++ b/deep_speech_2/infer.py @@ -189,7 +189,7 @@ def infer(): wer_counter += 1 print("cur wer = %f , average wer = %f" % (wer_cur, wer_sum / wer_counter)) - # beam search in multiple processes + # beam search using multiple processes elif args.decode_method == "beam_search_nproc": ext_scorer = Scorer(args.alpha, args.beta, args.language_model_path) beam_search_nproc_results = ctc_beam_search_decoder_nproc(