嘉渊
2023-05-11 decdbc3c97e51c9cc48b2b39f8b273aa1f5a01c8
funasr/models/encoder/transformer_encoder.py
@@ -37,7 +37,7 @@
from funasr.modules.subsampling import check_short_utt
class EncoderLayer(AbsEncoder):
class EncoderLayer(nn.Module):
    """Encoder layer module.
    Args:
@@ -144,7 +144,7 @@
        return x, mask
class TransformerEncoder(torch.nn.Module):
class TransformerEncoder(AbsEncoder):
    """Transformer encoder module.
    Args: