zhifu gao
2024-01-18 b28f3c9da94ae72a3a0b7bb5982b587be7cf4cd6
funasr/models/branchformer/encoder.py
@@ -43,7 +43,7 @@
    check_short_utt,
)
from funasr.utils.register import register_class
from funasr.register import tables
class BranchformerEncoderLayer(torch.nn.Module):
    """Branchformer encoder layer module.
@@ -291,7 +291,7 @@
        return x, mask
@register_class("encoder_classes", "BranchformerEncoder")
@tables.register("encoder_classes", "BranchformerEncoder")
class BranchformerEncoder(nn.Module):
    """Branchformer encoder module."""