modify unit test for speech_diarization_sond-en-us-callhome-8k-n16k4-pytorch

志浩

2023-03-09 c1f5bc2e4f9ba4f0ff10b96dca115b55c2e800de

1
2
3
4
5
6
7
8
9
10
11
12
13
14
from abc import ABC
from abc import abstractmethod
from typing import Tuple
 
import torch
 
 
class AbsNormalize(torch.nn.Module, ABC):
    @abstractmethod
    def forward(
        self, input: torch.Tensor, input_lengths: torch.Tensor = None
    ) -> Tuple[torch.Tensor, torch.Tensor]:
        # return output, output_lengths
        raise NotImplementedError