Skip to content

Commit e8a64fb

Browse files
committed
Test input size for efficientnet_v2s was wrong in last results run
1 parent a04427d commit e8a64fb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

timm/models/efficientnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,7 @@ def _cfg(url='', **kwargs):
156156

157157
'efficientnet_v2s': _cfg(
158158
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_v2s_ra2-b265c1ba.pth',
159-
input_size=(3, 224, 224), test_size=(3, 288, 288), pool_size=(7, 7), crop_pct=1.0), # FIXME WIP
159+
input_size=(3, 224, 224), test_input_size=(3, 320, 320), pool_size=(7, 7), crop_pct=1.0), # FIXME WIP
160160

161161
'tf_efficientnet_b0': _cfg(
162162
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b0_aa-827b6e33.pth',

0 commit comments

Comments
 (0)