(self, x)
| 214 | self.sigmoid = nn.Sigmoid() |
| 215 | |
| 216 | def forward(self, x): |
| 217 | avgout = self.shared_MLP(self.avg_pool(x)) |
| 218 | maxout = self.shared_MLP(self.max_pool(x)) |
| 219 | return self.sigmoid(avgout + maxout) |
| 220 | |
| 221 | |
| 222 | class SpatialAttentionModule(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected