Skip to content

Commit 65671dd

Browse files
committed
Fix static padding calculation
1 parent 5fbffa4 commit 65671dd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

efficientnet_pytorch/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -261,8 +261,8 @@ def __init__(self, in_channels, out_channels, kernel_size, stride=1, image_size=
261261
pad_h = max((oh - 1) * self.stride[0] + (kh - 1) * self.dilation[0] + 1 - ih, 0)
262262
pad_w = max((ow - 1) * self.stride[1] + (kw - 1) * self.dilation[1] + 1 - iw, 0)
263263
if pad_h > 0 or pad_w > 0:
264-
self.static_padding = nn.ZeroPad2d((pad_w - pad_w // 2, pad_w - pad_w // 2,
265-
pad_h - pad_h // 2, pad_h - pad_h // 2))
264+
self.static_padding = nn.ZeroPad2d((pad_w // 2, pad_w - pad_w // 2,
265+
pad_h // 2, pad_h - pad_h // 2))
266266
else:
267267
self.static_padding = nn.Identity()
268268

0 commit comments

Comments
 (0)