From 8f99a896eb1973452d6af3fb70efb040cebdce32 Mon Sep 17 00:00:00 2001 From: tachengP <2638591622@qq.com> Date: Fri, 17 Jul 2026 14:14:38 +0800 Subject: [PATCH] =?UTF-8?q?k=5Ffilter=E6=94=B9=E5=9B=9Enn.Parameter?= =?UTF-8?q?=E4=BB=A5=E5=8C=B9=E9=85=8DR3MOE=E6=A3=80=E6=9F=A5=E7=82=B9?= =?UTF-8?q?=E7=9A=84=E7=BB=93=E6=9E=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- modules/estimators/nets.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/modules/estimators/nets.py b/modules/estimators/nets.py index 5bd3debf2..fb3dd1280 100644 --- a/modules/estimators/nets.py +++ b/modules/estimators/nets.py @@ -61,10 +61,8 @@ def __init__( ) # post process - # self.k_filter = nn.Parameter(torch.ones(num_speakers)) + self.k_filter = nn.Parameter(torch.ones(num_speakers)) self.p0 = (0.0 - self.vmin) / (self.vmax - self.vmin) - self.k_filter = nn.Embedding(num_speakers, 1) - nn.init.constant_(self.k_filter.weight, 0.0) def forward(self, x: torch.Tensor, spk_id: torch.Tensor=None) -> torch.Tensor: """ @@ -78,7 +76,7 @@ def forward(self, x: torch.Tensor, spk_id: torch.Tensor=None) -> torch.Tensor: x, _ = self.rnn(x) x = self.output_proj(x) if spk_id is not None: - k_filter = self.k_filter(spk_id).view(-1,1,1) + k_filter = self.k_filter[spk_id].view(-1,1,1) k_filter = 0.5 + 1.5 * torch.sigmoid(k_filter) k_filter = torch.clamp(k_filter, min=0.3, max=2.5) x = (x - self.p0) * k_filter + self.p0