Просмотр исходного кода

update infer func for old contextual

shixian.shi 2 лет назад
Родитель
Сommit
5690a8bcb3
1 измененных файлов с 1 добавлено и 1 удалено
  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
         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:
         if hw_list is None:
             # default hotword list
             # default hotword list
             hw_list = [torch.Tensor([self.sos]).long().to(encoder_out.device)]  # empty hotword list
             hw_list = [torch.Tensor([self.sos]).long().to(encoder_out.device)]  # empty hotword list