onnx export funasr_onnx

游雁

2023-05-12 d25d0942f98ba5d879a2a39a8c30201c6496f3ae

 funasr/export/models/encoder/conformer_encoder.py

@@ -61,7 +61,6 @@
                speech: torch.Tensor,
                speech_lengths: torch.Tensor,
                ):
        speech = speech * self._output_size ** 0.5
        mask = self.make_pad_mask(speech_lengths)
        mask = self.prepare_mask(mask)
        if self.embed is None: