This commit is contained in:
嘉渊 2023-04-28 15:28:01 +08:00
parent ce2f775f44
commit e1549946bc

View File

@ -43,7 +43,7 @@ from funasr.modules.subsampling import TooShortUttError
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 @@ class EncoderLayer(nn.Module):
return x, mask
class ConformerEncoder(torch.nn.Module):
class ConformerEncoder(AbsEncoder):
"""Conformer encoder module.
Args: