amp pipeline

游雁

2023-02-16 6023271be65ddf888e26cffc646631809bf500a8

 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