(self, x, fencoder, mask)
| 291 | self.conv_out = nn.Conv1d(num_f_maps, num_classes, 1) |
| 292 | |
| 293 | def forward(self, x, fencoder, mask): |
| 294 | |
| 295 | feature = self.conv_1x1(x) |
| 296 | for layer in self.layers: |
| 297 | feature = layer(feature, fencoder, mask) |
| 298 | |
| 299 | out = self.conv_out(feature) * mask[:, 0:1, :] |
| 300 | |
| 301 | return out, feature |
| 302 | |
| 303 | class MyTransformer(nn.Module): |
| 304 | def __init__(self, num_decoders, num_layers, r1, r2, num_f_maps, input_dim, num_classes, channel_masking_rate): |
nothing calls this directly
no outgoing calls
no test coverage detected