From e04489ce4c0fd0095d0c79ef8f504f425e0435a8 Mon Sep 17 00:00:00 2001
From: Shi Xian <40013335+R1ckShi@users.noreply.github.com>
Date: 星期三, 13 三月 2024 16:34:42 +0800
Subject: [PATCH] contextual&seaco ONNX export (#1481)
---
funasr/models/sond/encoder/ci_scorers.py | 6 ------
1 files changed, 0 insertions(+), 6 deletions(-)
diff --git a/funasr/models/sond/encoder/ci_scorers.py b/funasr/models/sond/encoder/ci_scorers.py
index 50056ee..c5f45b9 100644
--- a/funasr/models/sond/encoder/ci_scorers.py
+++ b/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 {}
--
Gitblit v1.9.1