From 53fccccb24d15d788919d91c8c2b06a115ddacf3 Mon Sep 17 00:00:00 2001
From: 夜雨飘零 <yeyupiaoling@foxmail.com>
Date: 星期二, 19 十二月 2023 15:39:39 +0800
Subject: [PATCH] 修改cnn为合理的包名
---
funasr/models/specaug/specaug.py | 12 ++++++------
1 files changed, 6 insertions(+), 6 deletions(-)
diff --git a/funasr/models/specaug/specaug.py b/funasr/models/specaug/specaug.py
index 6074f86..e5da8e2 100644
--- a/funasr/models/specaug/specaug.py
+++ b/funasr/models/specaug/specaug.py
@@ -3,14 +3,14 @@
from typing import Sequence
from typing import Union
-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
+from funasr.models.specaug.mask_along_axis import MaskAlongAxis
+from funasr.models.specaug.mask_along_axis import MaskAlongAxisVariableMaxWidth
+from funasr.models.specaug.mask_along_axis import MaskAlongAxisLFR
+from funasr.models.specaug.time_warp import TimeWarp
+import torch.nn as nn
-class SpecAug(AbsSpecAug):
+class SpecAug(nn.Module):
"""Implementation of SpecAug.
Reference:
--
Gitblit v1.9.1