From 23bc5dee4e88ef8b5d0c0d64d2e188c054422e8b Mon Sep 17 00:00:00 2001
From: speech_asr <wangjiaming.wjm@alibaba-inc.com>
Date: 星期二, 11 四月 2023 00:21:45 +0800
Subject: [PATCH] update
---
/dev/null | 18 ------------------
funasr/models/specaug/specaug.py | 7 ++++---
2 files changed, 4 insertions(+), 21 deletions(-)
diff --git a/funasr/models/specaug/abs_specaug.py b/funasr/models/specaug/abs_specaug.py
deleted file mode 100644
index 3cbac41..0000000
--- a/funasr/models/specaug/abs_specaug.py
+++ /dev/null
@@ -1,18 +0,0 @@
-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
diff --git a/funasr/models/specaug/specaug.py b/funasr/models/specaug/specaug.py
index 6074f86..75c5bb2 100644
--- a/funasr/models/specaug/specaug.py
+++ b/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锛歭ow frame rate
"""
--
Gitblit v1.9.1