(self, x)
| 387 | # nn.init.constant_(m.bias, 0) |
| 388 | |
| 389 | def forward(self, x): |
| 390 | conv0 = self.conv0(x) |
| 391 | conv2 = self.conv2(self.conv1(conv0)) |
| 392 | conv4 = self.conv4(self.conv3(conv2)) |
| 393 | x = self.conv6(self.conv5(conv4)) |
| 394 | x = conv4 + self.conv7(x) |
| 395 | x = conv2 + self.conv9(x) |
| 396 | x = conv0 + self.conv11(x) |
| 397 | x = self.prob(x) |
| 398 | return x |
| 399 | |
| 400 | class CostRegNet_part_refine(nn.Module): |
| 401 | def __init__(self, in_channels, base_channels,stage=0): |
nothing calls this directly
no outgoing calls
no test coverage detected