querryton
2024-04-20 01df8f330ccc754223d5e2d688dc0a55d27f2dcc
funasr/models/sond/encoder/ci_scorers.py
@@ -16,9 +16,6 @@
        scores = torch.matmul(xs_pad, spk_emb.transpose(1, 2))
        return scores
    def convert_tf2torch(self, var_dict_tf, var_dict_torch):
        return {}
class CosScorer(torch.nn.Module):
    def __init__(self):
@@ -33,6 +30,3 @@
        # spk_emb: B, N, D
        scores = F.cosine_similarity(xs_pad.unsqueeze(2), spk_emb.unsqueeze(1), dim=-1)
        return scores
    def convert_tf2torch(self, var_dict_tf, var_dict_torch):
        return {}