From f701679677892df4f225226209e32a3320bafbee Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期五, 28 四月 2023 15:34:48 +0800
Subject: [PATCH] update
---
funasr/models/encoder/sanm_encoder.py | 10 ++++++----
1 files changed, 6 insertions(+), 4 deletions(-)
diff --git a/funasr/models/encoder/sanm_encoder.py b/funasr/models/encoder/sanm_encoder.py
index 1462403..8f30c1f 100644
--- a/funasr/models/encoder/sanm_encoder.py
+++ b/funasr/models/encoder/sanm_encoder.py
@@ -25,8 +25,10 @@
from funasr.modules.subsampling import Conv2dSubsampling8
from funasr.modules.subsampling import TooShortUttError
from funasr.modules.subsampling import check_short_utt
-from funasr.models.ctc import CTC
from funasr.modules.mask import subsequent_mask, vad_mask
+
+from funasr.models.ctc import CTC
+from funasr.models.encoder.abs_encoder import AbsEncoder
class EncoderLayerSANM(nn.Module):
def __init__(
@@ -114,7 +116,7 @@
return x, mask, cache, mask_shfit_chunk, mask_att_chunk_encoder
-class SANMEncoder(torch.nn.Module):
+class SANMEncoder(AbsEncoder):
"""
author: Speech Lab, Alibaba Group, China
San-m: Memory equipped self-attention for end-to-end speech recognition
@@ -546,7 +548,7 @@
return var_dict_torch_update
-class SANMEncoderChunkOpt(torch.nn.Module):
+class SANMEncoderChunkOpt(AbsEncoder):
"""
author: Speech Lab, Alibaba Group, China
SCAMA: Streaming chunk-aware multihead attention for online end-to-end speech recognition
@@ -959,7 +961,7 @@
return var_dict_torch_update
-class SANMVadEncoder(torch.nn.Module):
+class SANMVadEncoder(AbsEncoder):
"""
author: Speech Lab, Alibaba Group, China
--
Gitblit v1.9.1