From d5a80d642a5721eb1352cba59833a5cf4b91000f Mon Sep 17 00:00:00 2001
From: speech_asr <wangjiaming.wjm@alibaba-inc.com>
Date: 星期二, 11 四月 2023 00:09:29 +0800
Subject: [PATCH] update
---
funasr/models/encoder/sanm_encoder.py | 7 +++----
1 files changed, 3 insertions(+), 4 deletions(-)
diff --git a/funasr/models/encoder/sanm_encoder.py b/funasr/models/encoder/sanm_encoder.py
index 2a3a353..1462403 100644
--- a/funasr/models/encoder/sanm_encoder.py
+++ b/funasr/models/encoder/sanm_encoder.py
@@ -26,7 +26,6 @@
from funasr.modules.subsampling import TooShortUttError
from funasr.modules.subsampling import check_short_utt
from funasr.models.ctc import CTC
-from funasr.models.encoder.abs_encoder import AbsEncoder
from funasr.modules.mask import subsequent_mask, vad_mask
class EncoderLayerSANM(nn.Module):
@@ -115,7 +114,7 @@
return x, mask, cache, mask_shfit_chunk, mask_att_chunk_encoder
-class SANMEncoder(AbsEncoder):
+class SANMEncoder(torch.nn.Module):
"""
author: Speech Lab, Alibaba Group, China
San-m: Memory equipped self-attention for end-to-end speech recognition
@@ -547,7 +546,7 @@
return var_dict_torch_update
-class SANMEncoderChunkOpt(AbsEncoder):
+class SANMEncoderChunkOpt(torch.nn.Module):
"""
author: Speech Lab, Alibaba Group, China
SCAMA: Streaming chunk-aware multihead attention for online end-to-end speech recognition
@@ -960,7 +959,7 @@
return var_dict_torch_update
-class SANMVadEncoder(AbsEncoder):
+class SANMVadEncoder(torch.nn.Module):
"""
author: Speech Lab, Alibaba Group, China
--
Gitblit v1.9.1