File size: 470 Bytes
ad16788 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
from abc import ABC
from abc import abstractmethod
from collections import OrderedDict
from typing import Tuple
import torch
class AbsSeparator(torch.nn.Module, ABC):
@abstractmethod
def forward(
self,
input: torch.Tensor,
ilens: torch.Tensor,
) -> Tuple[Tuple[torch.Tensor], torch.Tensor, OrderedDict]:
raise NotImplementedError
@property
@abstractmethod
def num_spk(self):
raise NotImplementedError
|