From 6e69d784e4814c3dbe35e8f70c6cf4b920c8b20b Mon Sep 17 00:00:00 2001
From: 天地 <tiandiweizun@gmail.com>
Date: 星期三, 19 三月 2025 23:10:13 +0800
Subject: [PATCH] 1. bug fix:list(mean)和list(var)，由于mean和var是numpy，导致写入到文件的格式错误，参考上面的话，大概率是list(mean.tolist())，其实外层list没有必要 (#2437)
---
funasr/models/transformer/utils/lightconv2d.py | 14 +++++---------
1 files changed, 5 insertions(+), 9 deletions(-)
diff --git a/funasr/models/transformer/utils/lightconv2d.py b/funasr/models/transformer/utils/lightconv2d.py
index 294d232..60e5af5 100644
--- a/funasr/models/transformer/utils/lightconv2d.py
+++ b/funasr/models/transformer/utils/lightconv2d.py
@@ -50,9 +50,7 @@
self.act = nn.GLU()
# lightconv related
- self.weight = nn.Parameter(
- torch.Tensor(self.wshare, 1, kernel_size).uniform_(0, 1)
- )
+ self.weight = nn.Parameter(torch.Tensor(self.wshare, 1, kernel_size).uniform_(0, 1))
self.weight_f = nn.Parameter(torch.Tensor(1, 1, kernel_size).uniform_(0, 1))
self.use_bias = use_bias
if self.use_bias:
@@ -93,9 +91,9 @@
# convolution along frequency axis
weight_f = F.softmax(self.weight_f, dim=-1)
weight_f = F.dropout(weight_f, self.dropout_rate, training=self.training)
- weight_new = torch.zeros(
- B * T, 1, self.kernel_size, device=x.device, dtype=x.dtype
- ).copy_(weight_f)
+ weight_new = torch.zeros(B * T, 1, self.kernel_size, device=x.device, dtype=x.dtype).copy_(
+ weight_f
+ )
xf = F.conv1d(
x.view(1, B * T, C), weight_new, padding=self.padding_size, groups=B * T
).view(B, T, C)
@@ -107,9 +105,7 @@
self.kernel_mask = self.kernel_mask.to(x.device)
weight = weight.masked_fill(self.kernel_mask == 0.0, float("-inf"))
weight = F.softmax(weight, dim=-1)
- x = F.conv1d(x, weight, padding=self.padding_size, groups=self.wshare).view(
- B, C, T
- )
+ x = F.conv1d(x, weight, padding=self.padding_size, groups=self.wshare).view(B, C, T)
if self.use_bias:
x = x + self.bias.view(1, -1, 1)
x = x.transpose(1, 2) # B x T x C
--
Gitblit v1.9.1