From 28325e9dce4fdce36a38b3f6fc1e0475922814d2 Mon Sep 17 00:00:00 2001 From: zylo117 Date: Sun, 31 May 2020 18:46:02 +0800 Subject: [PATCH] Revert "Update utils_extra.py" --- efficientnet/utils_extra.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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