From fc968b01add44b2de36e9a013389170fa35ddedd Mon Sep 17 00:00:00 2001 From: Ubuntu Date: Wed, 21 Feb 2024 21:54:09 +0000 Subject: [PATCH] fixed multisoftmax for single language --- nemo/collections/asr/models/hybrid_rnnt_ctc_bpe_models.py | 1 - 1 file changed, 1 deletion(-) diff --git a/nemo/collections/asr/models/hybrid_rnnt_ctc_bpe_models.py b/nemo/collections/asr/models/hybrid_rnnt_ctc_bpe_models.py index 18dfe484d..87231e449 100644 --- a/nemo/collections/asr/models/hybrid_rnnt_ctc_bpe_models.py +++ b/nemo/collections/asr/models/hybrid_rnnt_ctc_bpe_models.py @@ -148,7 +148,6 @@ def __init__(self, cfg: DictConfig, trainer: Trainer = None): with open_dict(self.cfg.aux_ctc): self.cfg.aux_ctc.decoding = ctc_decoding_cfg if (self.tokenizer_type == "agg" or self.tokenizer_type == "multilingual") and "multisoftmax" in cfg.decoder: - breakpoint() if ctc_decoding_cfg.strategy == 'pyctcdecode': self.decoding = CTCBPEDecoding(self.cfg.decoding, tokenizer=self.tokenizer, blank_id=(self.ctc_decoder._num_classes-1)//len(self.tokenizer.tokenizers_dict.keys()),lang='any') else: