Update Time_TravelRephotography/model.py

Time_TravelRephotography/model.py

@@ -9,7 +9,7 @@ from torch import nn
 from torch.nn import functional as F
 from torch.autograd import Function
 
-#from op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d
+from op import FusedLeakyReLU#, fused_leaky_relu, upfirdn2d
 from torch_utils.ops.bias_act import *
 from torch_utils.ops.upfirdn2d import *
 
@@ -333,7 +333,7 @@ class StyledConv(nn.Module):
         self.noise = NoiseInjection()
         # self.bias = nn.Parameter(torch.zeros(1, out_channel, 1, 1))
         # self.activate = ScaledLeakyReLU(0.2)
-        self.activate = bias_act(out_channel)
+        self.activate = FusedLeakyReLU(out_channel)
 
     def forward(self, input, style, noise=None):
         out = self.conv(input, style)