Update Time_TravelRephotography/model.py

Time_TravelRephotography/model.py

@@ -47,7 +47,7 @@ class Upsample(nn.Module):
         self.pad = (pad0, pad1)
 
     def forward(self, input):
-        out = upfirdn2d(input, self.kernel, up=self.factor, down=1, pad=self.pad)
+        out = upfirdn2d(input, self.kernel, up=self.factor, down=1, padding=[self.pad[0], self.pad[1], self.pad[0], self.pad[1]])
 
         return out
 
@@ -68,7 +68,7 @@ class Downsample(nn.Module):
         self.pad = (pad0, pad1)
 
     def forward(self, input):
-        out = upfirdn2d(input, self.kernel, up=1, down=self.factor, pad=self.pad)
+        out = upfirdn2d(input, self.kernel, up=1, down=self.factor, padding=[self.pad[0], self.pad[1], self.pad[0], self.pad[1]])
 
         return out