diff --git a/efficientnet_pytorch/utils.py b/efficientnet_pytorch/utils.py index 826a627..29af0ed 100755 --- a/efficientnet_pytorch/utils.py +++ b/efficientnet_pytorch/utils.py @@ -242,7 +242,7 @@ def forward(self, x): pad_w = max((ow - 1) * self.stride[1] + (kw - 1) * self.dilation[1] + 1 - iw, 0) if pad_h > 0 or pad_w > 0: x = F.pad(x, [pad_w // 2, pad_w - pad_w // 2, pad_h // 2, pad_h - pad_h // 2]) - return F.conv2d(x, self.weight, self.bias, self.stride, self.padding, self.dilation, self.groups) + return F.conv2d(x, self.weight, self.bias, self.dilation, self.groups) class Conv2dStaticSamePadding(nn.Conv2d):