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
---
funasr/models/specaug/specaug.py | 7 ++++---
1 files changed, 4 insertions(+), 3 deletions(-)
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