雾聪
2023-11-29 6bbeacb91634fa7279e9d8b3d07eeb85b7e80cb4
funasr/models/specaug/abs_specaug.py
New file
@@ -0,0 +1,16 @@
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