mirror of
https://github.com/HumanAIGC/lite-avatar.git
synced 2026-02-05 01:49:19 +08:00
19 lines
426 B
Python
19 lines
426 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
|