(self, x)
| 23 | self.bn_out = nn.BatchNorm2d(out_channels) if config.batch_size > 1 else nn.Identity() |
| 24 | |
| 25 | def forward(self, x): |
| 26 | x = self.conv_in(x) |
| 27 | x = self.bn_in(x) |
| 28 | x = self.relu_in(x) |
| 29 | if hasattr(self, 'dec_att'): |
| 30 | x = self.dec_att(x) |
| 31 | x = self.conv_out(x) |
| 32 | x = self.bn_out(x) |
| 33 | return x |
| 34 | |
| 35 | |
| 36 | class ResBlk(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected