From ae49b2a8e1bc676e6014d8a12ebeec947b655e3e Mon Sep 17 00:00:00 2001
From: 莫拉古 <61447879+yechaoying@users.noreply.github.com>
Date: 星期五, 29 十一月 2024 09:55:43 +0800
Subject: [PATCH] 变量名写错了 (#2249)
---
funasr/models/sond/encoder/ci_scorers.py | 18 ++++++------------
1 files changed, 6 insertions(+), 12 deletions(-)
diff --git a/funasr/models/sond/encoder/ci_scorers.py b/funasr/models/sond/encoder/ci_scorers.py
index 50056ee..a60560b 100644
--- a/funasr/models/sond/encoder/ci_scorers.py
+++ b/funasr/models/sond/encoder/ci_scorers.py
@@ -7,17 +7,14 @@
super().__init__()
def forward(
- self,
- xs_pad: torch.Tensor,
- spk_emb: torch.Tensor,
+ self,
+ xs_pad: torch.Tensor,
+ spk_emb: torch.Tensor,
):
# xs_pad: B, T, D
# spk_emb: B, N, D
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):
@@ -25,14 +22,11 @@
super().__init__()
def forward(
- self,
- xs_pad: torch.Tensor,
- spk_emb: torch.Tensor,
+ self,
+ xs_pad: torch.Tensor,
+ spk_emb: torch.Tensor,
):
# xs_pad: B, T, D
# 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