Method forward

models/sr3_dwt.py:389–396

(self, x, cond)

        nn.init.zeros_(self.body[-1].bias)

    def forward(self, x, cond):
        cond = self.body(cond)
        scale, shift = cond.chunk(2, dim=1)

        x = self.x_conv(x)

        x = x * (1 + scale) + shift
        return x


class FreqCondInjection(nn.Module):