From 2e0b2086586212a2c04b5eb612a85796be1494f7 Mon Sep 17 00:00:00 2001
From: Dr. Artificial曾小健 <875100501@qq.com>
Date: 星期二, 24 十二月 2024 13:46:31 +0800
Subject: [PATCH] typo fix (#2330)
---
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