speech_asr
2023-04-11 23bc5dee4e88ef8b5d0c0d64d2e188c054422e8b
update
1个文件已修改
1个文件已删除
25 ■■■■ 已修改文件
funasr/models/specaug/abs_specaug.py 18 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/models/specaug/specaug.py 7 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/models/specaug/abs_specaug.py
File was deleted
funasr/models/specaug/specaug.py
@@ -3,14 +3,15 @@
from typing import Sequence
from typing import Union
from funasr.models.specaug.abs_specaug import AbsSpecAug
import torch.nn
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
class SpecAug(AbsSpecAug):
class SpecAug(torch.nn.Module):
    """Implementation of SpecAug.
    Reference:
@@ -99,7 +100,7 @@
            x, x_lengths = self.time_mask(x, x_lengths)
        return x, x_lengths
class SpecAugLFR(AbsSpecAug):
class SpecAugLFR(torch.nn.Module):
    """Implementation of SpecAug.
    lfr_rate:low frame rate
    """