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.upfirdn2d import 
 from torch_utils.ops.bias_act import bias_act,bias_act_relu
 from torch_utils.ops.upfirdn2d import upfirdn2d
 
@@ -89,7 +89,7 @@ class Blur(nn.Module):
         self.pad = pad
 
     def forward(self, input):
-        out = upfirdn2d(input, self.kernel, pad=self.pad)
+        out = upfirdn2d(input, self.kernel, padding=[self.pad[0], self.pad[1], self.pad[0], self.pad[1]])
 
         return out