From 713cfc1d79f5689b9bd89a3ed02a8ac8c93fca1f Mon Sep 17 00:00:00 2001
From: Quandwang <quandwang@hotmail.com>
Date: Thu, 21 Jul 2022 13:35:23 +0800
Subject: [PATCH] use original beam size

---
 egs/librispeech/ASR/conformer_ctc2/decode.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/egs/librispeech/ASR/conformer_ctc2/decode.py b/egs/librispeech/ASR/conformer_ctc2/decode.py
index 3abcf588e..7c659a698 100755
--- a/egs/librispeech/ASR/conformer_ctc2/decode.py
+++ b/egs/librispeech/ASR/conformer_ctc2/decode.py
@@ -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(