speech_asr
2023-03-15 fbec0f003d4de9e4b6ccb6bb58d2d4926a0ff332
funasr/models/e2e_diar_sond.py
@@ -342,6 +342,7 @@
        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)