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

modify unit test for speech_diarization_sond-en-us-callhome-8k-n16k4-pytorch

Browse Source
This commit is contained in:
志浩 2023-03-09 17:34:01 +08:00
parent b0abb2f4c0
commit bb40093a64
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
funasr/models/e2e_diar_sond.py
View File

@ -342,6 +342,7 @@ class DiarSondModel(AbsESPnetModel):
if isinstance(self.ci_scorer, AbsEncoder):
ci_simi = self.ci_scorer(ge_in, ge_len)[0]
ci_simi = torch.reshape(ci_simi, [bb, self.max_spk_num, tt]).permute([0, 2, 1])
else:
ci_simi = self.ci_scorer(speech_encoder_outputs, speaker_encoder_outputs)