mirror of
https://github.com/HumanAIGC/lite-avatar.git
synced 2026-02-05 01:49:19 +08:00
22 lines
503 B
Python
22 lines
503 B
Python
from abc import ABC
|
|
from abc import abstractmethod
|
|
from typing import Optional
|
|
from typing import Tuple
|
|
|
|
import torch
|
|
|
|
|
|
class AbsEncoder(torch.nn.Module, ABC):
|
|
@abstractmethod
|
|
def output_size(self) -> int:
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def forward(
|
|
self,
|
|
xs_pad: torch.Tensor,
|
|
ilens: torch.Tensor,
|
|
prev_states: torch.Tensor = None,
|
|
) -> Tuple[torch.Tensor, torch.Tensor, Optional[torch.Tensor]]:
|
|
raise NotImplementedError
|