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

志浩

2023-03-09 cf41c0aa3ac3f1a3517c2b3a1b6981386d4bc64b

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
from typing import Optional
from typing import Tuple
 
import torch
 
 
class AbsSpecAug(torch.nn.Module):
    """Abstract class for the augmentation of spectrogram
 
    The process-flow:
 
    Frontend  -> SpecAug -> Normalization -> Encoder -> Decoder
    """
 
    def forward(
        self, x: torch.Tensor, x_lengths: torch.Tensor = None
    ) -> Tuple[torch.Tensor, Optional[torch.Tensor]]:
        raise NotImplementedError