Merge pull request lukemelas#223 from matkalinowski/static_padding

static padding fixed
ulosc · Aug 25, 2020 · 14f973a · 14f973a
2 parents 6ca324c + fbd2ad7
commit 14f973a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/efficientnet_pytorch/utils.py b/efficientnet_pytorch/utils.py
@@ -261,7 +261,8 @@ def __init__(self, in_channels, out_channels, kernel_size, stride=1, image_size=
         pad_h = max((oh - 1) * self.stride[0] + (kh - 1) * self.dilation[0] + 1 - ih, 0)
         pad_w = max((ow - 1) * self.stride[1] + (kw - 1) * self.dilation[1] + 1 - iw, 0)
         if pad_h > 0 or pad_w > 0:
-            self.static_padding = nn.ZeroPad2d((pad_w // 2, pad_w - pad_w // 2, pad_h // 2, pad_h - pad_h // 2))
+            self.static_padding = nn.ZeroPad2d((pad_w - pad_w // 2, pad_w - pad_w // 2,
+                                                pad_h - pad_h // 2, pad_h - pad_h // 2))
         else:
             self.static_padding = nn.Identity()