keep empty speech result (#1772)

彭震东

2024-05-30 a64b7d8d8aeb2bb543ca703045a45f42470e9a63

 funasr/models/sond/label_aggregation.py

@@ -90,9 +90,7 @@
        self.hop_length = hop_length

    def extra_repr(self):
        return (
            f"hop_length={self.hop_length}, "
        )
        return f"hop_length={self.hop_length}, "

    def forward(
        self, input: torch.Tensor, ilens: torch.Tensor = None
@@ -107,7 +105,9 @@

        """

        output = F.max_pool1d(input.transpose(1, 2), self.hop_length, self.hop_length).transpose(1, 2)
        output = F.max_pool1d(input.transpose(1, 2), self.hop_length, self.hop_length).transpose(
            1, 2
        )
        olens = ilens // self.hop_length

        return output.to(input.dtype), olens
        return output.to(input.dtype), olens