Merge pull request #223 from matkalinowski/static_padding

static padding fixed

Author: lukemelas

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()