From e8a64fb88108b592da192e98054095b1ee25e96e Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sat, 17 Apr 2021 16:17:41 -0700 Subject: [PATCH] Test input size for efficientnet_v2s was wrong in last results run --- timm/models/efficientnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/efficientnet.py b/timm/models/efficientnet.py index f77ab2ee..8f285759 100644 --- a/timm/models/efficientnet.py +++ b/timm/models/efficientnet.py @@ -156,7 +156,7 @@ default_cfgs = { 'efficientnet_v2s': _cfg( url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_v2s_ra2-b265c1ba.pth', - input_size=(3, 224, 224), test_size=(3, 288, 288), pool_size=(7, 7), crop_pct=1.0), # FIXME WIP + input_size=(3, 224, 224), test_input_size=(3, 320, 320), pool_size=(7, 7), crop_pct=1.0), # FIXME WIP 'tf_efficientnet_b0': _cfg( url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b0_aa-827b6e33.pth',