speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update infer func for old contextual

Browse Source
This commit is contained in:
shixian.shi 2023-07-06 18:53:09 +08:00
parent 9321718b3a
commit 5690a8bcb3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/models/e2e_asr_paraformer.py
View File

@ -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