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
|
|
