(self, x)
| 210 | setattr(self, 'prelu{}'.format(i + 1), nn.PReLU(self.in_channels)) |
| 211 | |
| 212 | def forward(self, x): |
| 213 | skip = x |
| 214 | for i in range(self.depth): |
| 215 | out = getattr(self, 'pad{}'.format(i + 1))(skip) |
| 216 | out = getattr(self, 'conv{}'.format(i + 1))(out) |
| 217 | out = getattr(self, 'norm{}'.format(i + 1))(out) |
| 218 | out = getattr(self, 'prelu{}'.format(i + 1))(out) |
| 219 | skip = out |
| 220 | return out |
| 221 | |
| 222 | class DFNet(nn.Module): |
| 223 | def __init__(self, width=64, input_channel_rate=1, output_channel=2): |
nothing calls this directly
no outgoing calls
no test coverage detected