Update model.py

williamyang1991 · Sep 19, 2023 · e593d16 · e593d16
1 parent 60ba4bd
commit e593d16
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/models/stylegan2/model.py b/models/stylegan2/model.py
@@ -230,7 +230,7 @@ def __init__(
 
         fan_in = in_channel * kernel_size ** 2
         self.scale = 1 / math.sqrt(fan_in)
-        self.padding = kernel_size // 2 + dilation - 1 ##### modified
+        self.padding = (kernel_size - 1) // 2 * dilation ##### modified
 
         self.weight = nn.Parameter(
             torch.randn(1, out_channel, in_channel, kernel_size, kernel_size)
@@ -657,7 +657,7 @@ def __init__(
 
         else:
             stride = 1
-            self.padding = kernel_size // 2 + dilation-1 ## modified
+            self.padding = (kernel_size - 1) // 2 * dilation ## modified
 
         layers.append(
             EqualConv2d(
@@ -765,4 +765,4 @@ def forward(self, input):
         out = out.view(batch, -1)
         out = self.final_linear(out)
 
-        return out
+        return out