Skip to content

Commit

Permalink
fixed multisoftmax for single language
Browse files Browse the repository at this point in the history
  • Loading branch information
tahirjmakhdoomi committed Feb 21, 2024
1 parent 438aa07 commit fc968b0
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion nemo/collections/asr/models/hybrid_rnnt_ctc_bpe_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit fc968b0

Please sign in to comment.