1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
| from abc import ABC
| from abc import abstractmethod
| from typing import Tuple
|
| import torch
|
| from funasr.modules.scorers.scorer_interface import ScorerInterface
|
|
| class AbsDecoder(torch.nn.Module, ScorerInterface, ABC):
| @abstractmethod
| def forward(
| self,
| hs_pad: torch.Tensor,
| hlens: torch.Tensor,
| ys_in_pad: torch.Tensor,
| ys_in_lens: torch.Tensor,
| ) -> Tuple[torch.Tensor, torch.Tensor]:
| raise NotImplementedError
|
|
