amp pipeline

游雁

2023-02-16 b841635a53427f306886af8a15d89aef7163727b

 funasr/punctuation/abs_model.py

@@ -23,7 +23,5 @@
    """

    @abstractmethod
    def forward(
        self, input: torch.Tensor, hidden: torch.Tensor
    ) -> Tuple[torch.Tensor, torch.Tensor]:
    def forward(self, input: torch.Tensor, hidden: torch.Tensor) -> Tuple[torch.Tensor, torch.Tensor]:
        raise NotImplementedError