update

嘉渊

2023-04-28 f701679677892df4f225226209e32a3320bafbee

update

 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