From 4fbc32d3d073b48860c8e6603b76d1becc7f4af1 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Tue, 11 May 2021 10:50:15 -0700 Subject: [PATCH] Fix crop_pct for cait models. --- timm/models/cait.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/cait.py b/timm/models/cait.py index b648e712..c5f7742f 100644 --- a/timm/models/cait.py +++ b/timm/models/cait.py @@ -26,7 +26,7 @@ def _cfg(url='', **kwargs): return { 'url': url, 'num_classes': 1000, 'input_size': (3, 384, 384), 'pool_size': None, - 'crop_pct': .9, 'interpolation': 'bicubic', 'fixed_input_size': True, + 'crop_pct': 1.0, 'interpolation': 'bicubic', 'fixed_input_size': True, 'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD, 'first_conv': 'patch_embed.proj', 'classifier': 'head', **kwargs