From 60696d3eb26be22ef42ab9df0ce109ac7e76ac4a Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Fri, 24 Dec 2021 11:16:29 +0800 Subject: [PATCH] Minor fixes. --- .../ASR/transducer_stateless/pretrained.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/transducer_stateless/pretrained.py b/egs/librispeech/ASR/transducer_stateless/pretrained.py index 49efa6749..6a6626371 100755 --- a/egs/librispeech/ASR/transducer_stateless/pretrained.py +++ b/egs/librispeech/ASR/transducer_stateless/pretrained.py @@ -110,6 +110,15 @@ def get_parser(): help="Used only when --method is beam_search", ) + parser.add_argument( + "--max-sym-per-frame", + type=int, + default=3, + help="""Maximum number of symbols per frame. Used only when + --method is greedy_search. + """, + ) + return parser @@ -279,7 +288,11 @@ def main(): encoder_out_i = encoder_out[i:i+1, :encoder_out_lens[i]] # fmt: on if params.method == "greedy_search": - hyp = greedy_search(model=model, encoder_out=encoder_out_i) + hyp = greedy_search( + model=model, + encoder_out=encoder_out_i, + max_sym_per_frame=params.max_sym_per_frame, + ) elif params.method == "beam_search": hyp = beam_search( model=model, encoder_out=encoder_out_i, beam=params.beam_size