(self, input, input_mask)
| 124 | self.activation = nn.LeakyReLU(negative_slope=0.2) |
| 125 | |
| 126 | def forward(self, input, input_mask): |
| 127 | h, h_mask = self.conv(input, input_mask) |
| 128 | if hasattr(self, 'bn'): |
| 129 | h = self.bn(h) |
| 130 | if hasattr(self, 'activation'): |
| 131 | h = self.activation(h) |
| 132 | return h, h_mask |
| 133 | |
| 134 | class Inpaint_Depth_Net(nn.Module): |
| 135 | def __init__(self, layer_size=7, upsampling_mode='nearest'): |
nothing calls this directly
no outgoing calls
no test coverage detected