Update Time_TravelRephotography/model.py

Time_TravelRephotography/model.py

@@ -9,9 +9,9 @@ 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 torch_utils.ops.bias_act import bias_act
-from torch_utils.ops.upfirdn2d import *
+#from op import FusedLeakyReLU, fused_leaky_relu#, upfirdn2d
+from torch_utils.ops.bias_act import bias_act,bias_act_relu
+from torch_utils.ops.upfirdn2d import upfirdn2d
 
 
 class PixelNorm(nn.Module):
@@ -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_ref(out_channel)
+        self.activate =bias_act_relu(out_channel)
 
     def forward(self, input, style, noise=None):
         out = self.conv(input, style)