Source code for espnet2.diar.abs_diar

from abc import ABC, abstractmethod
from collections import OrderedDict
from typing import Tuple

import torch


[docs]class AbsDiarization(torch.nn.Module, ABC): # @abstractmethod # def output_size(self) -> int: # raise NotImplementedError
[docs] @abstractmethod def forward( self, input: torch.Tensor, ilens: torch.Tensor, ) -> Tuple[torch.Tensor, torch.Tensor, OrderedDict]: raise NotImplementedError
[docs] @abstractmethod def forward_rawwav( self, input: torch.Tensor, ilens: torch.Tensor ) -> Tuple[torch.Tensor, torch.Tensor, OrderedDict]: raise NotImplementedError