From bb40093a641bab03d7d33e80702cafe67756b96f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BF=97=E6=B5=A9?= Date: Thu, 9 Mar 2023 17:34:01 +0800 Subject: [PATCH] modify unit test for speech_diarization_sond-en-us-callhome-8k-n16k4-pytorch --- funasr/models/e2e_diar_sond.py | 1 + 1 file changed, 1 insertion(+) diff --git a/funasr/models/e2e_diar_sond.py b/funasr/models/e2e_diar_sond.py index 419c8133a..258d78080 100644 --- a/funasr/models/e2e_diar_sond.py +++ b/funasr/models/e2e_diar_sond.py @@ -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)