(self, x, z)
| 137 | self.norm_s = ADAIN(input_nc, feature_nc) |
| 138 | |
| 139 | def forward(self, x, z): |
| 140 | x_s = self.shortcut(x, z) |
| 141 | dx = self.conv_0(self.actvn(self.norm_0(x, z))) |
| 142 | dx = self.conv_1(self.actvn(self.norm_1(dx, z))) |
| 143 | out = x_s + dx |
| 144 | return out |
| 145 | |
| 146 | def shortcut(self, x, z): |
| 147 | x_s = self.conv_s(self.actvn(self.norm_s(x, z))) |