嘉渊
2023-05-15 eb43576ed00902a5c0d5c05f5b50f9eebda3a0e1
funasr/models/encoder/conformer_encoder.py
@@ -43,7 +43,7 @@
from funasr.modules.subsampling import check_short_utt
from funasr.modules.subsampling import Conv2dSubsamplingPad
class ConvolutionModule(AbsEncoder):
class ConvolutionModule(nn.Module):
    """ConvolutionModule in Conformer model.
    Args:
@@ -278,7 +278,7 @@
        return x, mask
class ConformerEncoder(torch.nn.Module):
class ConformerEncoder(AbsEncoder):
    """Conformer encoder module.
    Args: