嘉渊
2023-04-28 f97e0eb9eee3f14c410ce905b73d0c83033dc1c9
funasr/models/encoder/conformer_encoder.py
@@ -19,6 +19,7 @@
    RelPositionMultiHeadedAttention,  # noqa: H301
    LegacyRelPositionMultiHeadedAttention,  # noqa: H301
)
from funasr.models.encoder.abs_encoder import AbsEncoder
from funasr.modules.embedding import (
    PositionalEncoding,  # noqa: H301
    ScaledPositionalEncoding,  # noqa: H301
@@ -41,7 +42,8 @@
from funasr.modules.subsampling import TooShortUttError
from funasr.modules.subsampling import check_short_utt
from funasr.modules.subsampling import Conv2dSubsamplingPad
class ConvolutionModule(nn.Module):
class ConvolutionModule(AbsEncoder):
    """ConvolutionModule in Conformer model.
    Args: