(self, dilation, in_channels, out_channels, r1, r2, att_type, stage, alpha)
| 213 | |
| 214 | class AttModule(nn.Module): |
| 215 | def __init__(self, dilation, in_channels, out_channels, r1, r2, att_type, stage, alpha): |
| 216 | super(AttModule, self).__init__() |
| 217 | self.feed_forward = ConvFeedForward(dilation, in_channels, out_channels) |
| 218 | self.instance_norm = nn.InstanceNorm1d(in_channels, track_running_stats=False) |
| 219 | self.att_layer = AttLayer(in_channels, in_channels, out_channels, r1, r1, r2, dilation, att_type=att_type, stage=stage) # dilation |
| 220 | self.conv_1x1 = nn.Conv1d(out_channels, out_channels, 1) |
| 221 | self.dropout = nn.Dropout() |
| 222 | self.alpha = alpha |
| 223 | |
| 224 | def forward(self, x, f, mask): |
| 225 | out = self.feed_forward(x) |
nothing calls this directly
no test coverage detected