Jelajahi Sumber

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

志浩 3 tahun lalu
induk
melakukan
bb40093a64
1 mengubah file dengan 1 tambahan dan 0 penghapusan
  1. 1 0
      funasr/models/e2e_diar_sond.py

+ 1 - 0
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)