From 81a5b29804800a4edd76c8dda2727d6fdf4b5643 Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期四, 11 五月 2023 17:35:49 +0800
Subject: [PATCH] update repo
---
funasr/models/specaug/specaug.py | 7 +++----
1 files changed, 3 insertions(+), 4 deletions(-)
diff --git a/funasr/models/specaug/specaug.py b/funasr/models/specaug/specaug.py
index 75c5bb2..6074f86 100644
--- a/funasr/models/specaug/specaug.py
+++ b/funasr/models/specaug/specaug.py
@@ -3,15 +3,14 @@
from typing import Sequence
from typing import Union
-import torch.nn
-
+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
-class SpecAug(torch.nn.Module):
+class SpecAug(AbsSpecAug):
"""Implementation of SpecAug.
Reference:
@@ -100,7 +99,7 @@
x, x_lengths = self.time_mask(x, x_lengths)
return x, x_lengths
-class SpecAugLFR(torch.nn.Module):
+class SpecAugLFR(AbsSpecAug):
"""Implementation of SpecAug.
lfr_rate锛歭ow frame rate
"""
--
Gitblit v1.9.1