diff --git a/efficientnet/utils_extra.py b/efficientnet/utils_extra.py index b9a6ffa76..9b4227c0b 100644 --- a/efficientnet/utils_extra.py +++ b/efficientnet/utils_extra.py @@ -80,7 +80,7 @@ def forward(self, x): top = extra_v // 2 bottom = extra_v - top - x = F.pad(x, [left, right, top, bottom], value=-float("inf")) + x = F.pad(x, [left, right, top, bottom]) x = self.pool(x) return x