Jelajahi Sumber

Merge branch 'main' of github.com:alibaba-damo-academy/FunASR
add

游雁 2 tahun lalu
induk
melakukan
b3f3ceefd7
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  1. 1 1
      funasr/models/e2e_asr_paraformer.py

+ 1 - 1
funasr/models/e2e_asr_paraformer.py

@@ -2107,7 +2107,7 @@ class ContextualParaformer(Paraformer):
 
         return loss_att, acc_att, cer_att, wer_att, loss_pre
 
-    def cal_decoder_with_predictor(self, encoder_out, encoder_out_lens, sematic_embeds, ys_pad_lens, hw_list=None):
+    def cal_decoder_with_predictor(self, encoder_out, encoder_out_lens, sematic_embeds, ys_pad_lens, hw_list=None, clas_scale=1.0):
         if hw_list is None:
             # default hotword list
             hw_list = [torch.Tensor([self.sos]).long().to(encoder_out.device)]  # empty hotword list