Method
__init__
(self, attn_layers, conv_layers=None, norm_layer=None)
Source from the content-addressed store, hash-verified
| 52 | |
| 53 | class Encoder(nn.Module): |
| 54 | def __init__(self, attn_layers, conv_layers=None, norm_layer=None): |
| 55 | super(Encoder, self).__init__() |
| 56 | self.attn_layers = nn.ModuleList(attn_layers) |
| 57 | self.conv_layers = nn.ModuleList(conv_layers) if conv_layers is not None else None |
| 58 | self.norm = norm_layer |
| 59 | |
| 60 | def forward(self, x, attn_mask=None): |
| 61 | # x [B, L, D] |
Tested by
no test coverage detected