(self, in_channel, style_dim, upsample=True, blur_kernel=[1, 3, 3, 1])
| 384 | |
| 385 | class ToFlow(nn.Module): |
| 386 | def __init__(self, in_channel, style_dim, upsample=True, blur_kernel=[1, 3, 3, 1]): |
| 387 | super().__init__() |
| 388 | |
| 389 | if upsample: |
| 390 | self.upsample = Upsample(blur_kernel) |
| 391 | |
| 392 | self.conv = ModulatedConv2d(in_channel, 3, 1, style_dim, demodulate=False) |
| 393 | self.bias = nn.Parameter(torch.zeros(1, 3, 1, 1)) |
| 394 | |
| 395 | def forward(self, input, style, feat, skip=None): |
| 396 | out = self.conv(input, style) |
nothing calls this directly
no test coverage detected