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/lightconv.py | 8 ++------
1 files changed, 2 insertions(+), 6 deletions(-)
diff --git a/funasr/models/transformer/utils/lightconv.py b/funasr/models/transformer/utils/lightconv.py
index b249402..b6d131b 100644
--- a/funasr/models/transformer/utils/lightconv.py
+++ b/funasr/models/transformer/utils/lightconv.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.use_bias = use_bias
if self.use_bias:
self.bias = nn.Parameter(torch.Tensor(n_feat))
@@ -96,9 +94,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