Spaces:
Runtime error
Runtime error
Update Time_TravelRephotography/model.py
Browse files
Time_TravelRephotography/model.py
CHANGED
@@ -333,7 +333,7 @@ class StyledConv(nn.Module):
|
|
333 |
self.noise = NoiseInjection()
|
334 |
# self.bias = nn.Parameter(torch.zeros(1, out_channel, 1, 1))
|
335 |
# self.activate = ScaledLeakyReLU(0.2)
|
336 |
-
self.activate =
|
337 |
|
338 |
def forward(self, input, style, noise=None):
|
339 |
out = self.conv(input, style)
|
|
|
333 |
self.noise = NoiseInjection()
|
334 |
# self.bias = nn.Parameter(torch.zeros(1, out_channel, 1, 1))
|
335 |
# self.activate = ScaledLeakyReLU(0.2)
|
336 |
+
self.activate =_bias_act_ref(out_channel)
|
337 |
|
338 |
def forward(self, input, style, noise=None):
|
339 |
out = self.conv(input, style)
|