From 2330e58f5ffa6d78cbdc099ac1a7e0aa1bd6a09f Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期四, 10 十月 2024 15:32:08 +0800
Subject: [PATCH] bugfix v1.1.11
---
funasr/models/transformer/utils/vgg2l.py | 8 ++------
1 files changed, 2 insertions(+), 6 deletions(-)
diff --git a/funasr/models/transformer/utils/vgg2l.py b/funasr/models/transformer/utils/vgg2l.py
index fb45d0b..7466bac 100644
--- a/funasr/models/transformer/utils/vgg2l.py
+++ b/funasr/models/transformer/utils/vgg2l.py
@@ -39,9 +39,7 @@
else:
self.output = torch.nn.Linear(128 * ((idim // 2) // 2), odim)
- def forward(
- self, feats: torch.Tensor, feats_mask: torch.Tensor
- ) -> Union[
+ def forward(self, feats: torch.Tensor, feats_mask: torch.Tensor) -> Union[
Tuple[torch.Tensor, torch.Tensor],
Tuple[Tuple[torch.Tensor, torch.Tensor], torch.Tensor],
]:
@@ -62,9 +60,7 @@
b, c, t, f = vgg_output.size()
- vgg_output = self.output(
- vgg_output.transpose(1, 2).contiguous().view(b, t, c * f)
- )
+ vgg_output = self.output(vgg_output.transpose(1, 2).contiguous().view(b, t, c * f))
if feats_mask is not None:
vgg_mask = self.create_new_mask(feats_mask)
--
Gitblit v1.9.1