amp pipeline

游雁

2023-02-16 b841635a53427f306886af8a15d89aef7163727b

amp pipeline

 funasr/bin/asr_inference_paraformer.py

@@ -337,7 +337,7 @@
        # assert check_return_type(results)
        return results

class Speech2TextExport(torch.nn.Module):
class Speech2TextExport:
    """Speech2TextExport class

    """
@@ -416,7 +416,7 @@
        self.asr_model = model
        
    @torch.no_grad()
    def forward(
    def __call__(
            self, speech: Union[torch.Tensor, np.ndarray], speech_lengths: Union[torch.Tensor, np.ndarray] = None
    ):
        """Inference