From 1596f6f414f6f41da66506debb1dff19fffeb3ec Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 24 六月 2024 11:55:17 +0800
Subject: [PATCH] fixbug hotwords
---
funasr/models/transformer/utils/dynamic_conv2d.py | 10 ++++------
1 files changed, 4 insertions(+), 6 deletions(-)
diff --git a/funasr/models/transformer/utils/dynamic_conv2d.py b/funasr/models/transformer/utils/dynamic_conv2d.py
index f8a4dd6..29d4dd8 100644
--- a/funasr/models/transformer/utils/dynamic_conv2d.py
+++ b/funasr/models/transformer/utils/dynamic_conv2d.py
@@ -95,9 +95,7 @@
# convolution of frequency axis
weight_f = self.linear_weight_f(x).view(B * T, 1, k) # B x T x k
self.attn_f = weight_f.view(B, T, k).unsqueeze(1)
- xf = F.conv1d(
- x.view(1, B * T, C), weight_f, padding=self.padding_size, groups=B * T
- )
+ xf = F.conv1d(x.view(1, B * T, C), weight_f, padding=self.padding_size, groups=B * T)
xf = xf.view(B, T, C)
# get kernel of convolution
@@ -107,9 +105,9 @@
weight_new = torch.zeros(B * H * T * (T + k - 1), dtype=weight.dtype)
weight_new = weight_new.view(B, H, T, T + k - 1).fill_(float("-inf"))
weight_new = weight_new.to(x.device) # B x H x T x T+k-1
- weight_new.as_strided(
- (B, H, T, k), ((T + k - 1) * T * H, (T + k - 1) * T, T + k, 1)
- ).copy_(weight)
+ weight_new.as_strided((B, H, T, k), ((T + k - 1) * T * H, (T + k - 1) * T, T + k, 1)).copy_(
+ weight
+ )
weight_new = weight_new.narrow(-1, int((k - 1) / 2), T) # B x H x T x T(k)
if self.use_kernel_mask:
kernel_mask = torch.tril(torch.ones(T, T, device=x.device)).unsqueeze(0)
--
Gitblit v1.9.1