From 6ed27c64c96c6f8b148c6d4110716cba6a185452 Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期四, 27 四月 2023 17:19:39 +0800
Subject: [PATCH] update

---
 funasr/models/specaug/specaug.py |    5 ++---
 1 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/funasr/models/specaug/specaug.py b/funasr/models/specaug/specaug.py
index 75c5bb2..d4e305d 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:

--
Gitblit v1.9.1