mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-19 05:54:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

use original beam size

Browse Source
This commit is contained in:
Quandwang 2022-07-21 13:35:23 +08:00
parent d812f82ea5
commit 713cfc1d79
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
egs/librispeech/ASR/conformer_ctc2/decode.py
View File

@ -201,8 +201,8 @@ def get_params() -> AttributeDict:
"encoder_dim": 512,
"num_encoder_layers": 12,
# parameters for decoding
"search_beam": 4,
"output_beam": 4,
"search_beam": 20,
"output_beam": 8,
"min_active_states": 30,
"max_active_states": 10000,
"use_double_scores": True,
@ -672,6 +672,10 @@ def main():
sos_id = graph_compiler.sos_id
eos_id = graph_compiler.eos_id
params.num_classes = num_classes
params.sos_id = sos_id
params.eos_id = eos_id
if params.method == "ctc-decoding" or params.method == "ctc-greedy-search":
HLG = None
H = k2.ctc_topo(