Merge branch 'dev_gzf' of github.com:alibaba-damo-academy/FunASR into dev_g...

游雁

2024-02-22 aaf61bbb6427aa50095f7682d3c0ef64b58b09bb

 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."""