mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
17 lines
424 B
Python
17 lines
424 B
Python
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
|