夜雨飘零
2023-12-19 53fccccb24d15d788919d91c8c2b06a115ddacf3
funasr/models/specaug/specaug.py
@@ -3,14 +3,14 @@
from typing import Sequence
from typing import Union
from funasr.models.specaug.abs_specaug import AbsSpecAug
from funasr.layers.mask_along_axis import MaskAlongAxis
from funasr.layers.mask_along_axis import MaskAlongAxisVariableMaxWidth
from funasr.layers.mask_along_axis import MaskAlongAxisLFR
from funasr.layers.time_warp import TimeWarp
from funasr.models.specaug.mask_along_axis import MaskAlongAxis
from funasr.models.specaug.mask_along_axis import MaskAlongAxisVariableMaxWidth
from funasr.models.specaug.mask_along_axis import MaskAlongAxisLFR
from funasr.models.specaug.time_warp import TimeWarp
import torch.nn as nn
class SpecAug(AbsSpecAug):
class SpecAug(nn.Module):
    """Implementation of SpecAug.
    Reference: