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/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