Change default (NVIDIA#8371) (NVIDIA#8372)

Signed-off-by: smajumdar <[email protected]> Co-authored-by: Somshubra Majumdar <[email protected]> Signed-off-by: biscayan <[email protected]>
biscayan · Feb 15, 2024 · c74fec5 · c74fec5
1 parent b23cb91
commit c74fec5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/asr/parts/submodules/multitask_decoding.py b/nemo/collections/asr/parts/submodules/multitask_decoding.py
@@ -486,7 +486,7 @@ class MultiTaskDecodingConfig:
     # )
 
     # beam decoding config
-    beam: AEDBeamInferConfig = field(default_factory=lambda: AEDBeamInferConfig(beam_size=4))
+    beam: AEDBeamInferConfig = field(default_factory=lambda: AEDBeamInferConfig(beam_size=1))
 
     # can be used to change temperature for decoding
     temperature: float = 1.0