mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
18 lines
403 B
Python
18 lines
403 B
Python
from abc import ABC
|
|
from abc import abstractmethod
|
|
from typing import Tuple
|
|
|
|
import torch
|
|
|
|
|
|
class AbsPostEncoder(torch.nn.Module, ABC):
|
|
@abstractmethod
|
|
def output_size(self) -> int:
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def forward(
|
|
self, input: torch.Tensor, input_lengths: torch.Tensor
|
|
) -> Tuple[torch.Tensor, torch.Tensor]:
|
|
raise NotImplementedError
|