working

hainan-xv · Jun 9, 2024 · d7a63bf · d7a63bf
1 parent 0daf523
commit d7a63bf
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/nemo/collections/asr/parts/submodules/ctc_decoding.py b/nemo/collections/asr/parts/submodules/ctc_decoding.py
@@ -213,6 +213,9 @@ def __init__(self, decoding_cfg, blank_id: int):
         self.batch_dim_index = self.cfg.get('batch_dim_index', 0)
         self.word_seperator = self.cfg.get('word_seperator', ' ')
 
+        if self.cfg.strategy == 'greedy_batched':
+            self.cfg.strategy = 'greedy_batch'
+
         possible_strategies = ['greedy', 'greedy_batch', 'beam', 'pyctcdecode', 'flashlight']
         if self.cfg.strategy not in possible_strategies:
             raise ValueError(f"Decoding strategy must be one of {possible_strategies}. Given {self.cfg.strategy}")