update

嘉渊

2023-04-28 7f74ab462e9a10a203dc5a9707a605198347365e

update

 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: