fixbug hotwords

游雁

2024-06-24 1596f6f414f6f41da66506debb1dff19fffeb3ec

 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)