update

嘉渊

2023-04-27 6ed27c64c96c6f8b148c6d4110716cba6a185452

 funasr/models/encoder/transformer_encoder.py

@@ -13,6 +13,7 @@
import logging

from funasr.models.ctc import CTC
from funasr.models.encoder.abs_encoder import AbsEncoder
from funasr.modules.attention import MultiHeadedAttention
from funasr.modules.embedding import PositionalEncoding
from funasr.modules.layer_norm import LayerNorm
@@ -36,7 +37,7 @@
from funasr.modules.subsampling import check_short_utt


class EncoderLayer(nn.Module):
class EncoderLayer(AbsEncoder):
    """Encoder layer module.

    Args: