Fix the variable name (#2328)

majic31

2024-12-24 23e7ddebccd3b05cf7ef89809bcfe565ad6dfa1f

 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)