Skip to content

Commit bda987b

Browse files
committed
Validation default cfg
1 parent 9dc7959 commit bda987b

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

timm/models/cvt.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -495,9 +495,9 @@ def _cfg(url='', **kwargs):
495495
return {
496496
'url': url,
497497
'num_classes': 1000, 'input_size': (3, 224, 224), 'pool_size': (14, 14),
498-
'crop_pct': 0.95, 'interpolation': 'bicubic',
498+
'crop_pct': 0.875, 'interpolation': 'bicubic',
499499
'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,
500-
'first_conv': 'stages.0.conv_embed.conv', 'classifier': 'head',
500+
'first_conv': 'stage0.patch_embed.proj', 'classifier': 'head',
501501
**kwargs
502502
}
503503

@@ -506,18 +506,18 @@ def _cfg(url='', **kwargs):
506506
url='https://github.com/fffffgggg54/pytorch-image-models/releases/download/cvt/CvT-13-224x224-IN-1k.pth'),
507507
'cvt_13.msft_in1k_384': _cfg(
508508
url='https://github.com/fffffgggg54/pytorch-image-models/releases/download/cvt/CvT-13-384x384-IN-1k.pth',
509-
input_size=(3, 384, 384), pool_size=(24, 24)),
509+
input_size=(3, 384, 384), pool_size=(24, 24), crop_mode='squash', crop_pct=1.0),
510510
'cvt_13.msft_in22k_ft_in1k_384': _cfg(url='https://github.com/fffffgggg54/pytorch-image-models/releases/download/cvt/CvT-13-384x384-IN-22k.pth',
511-
input_size=(3, 384, 384), pool_size=(24, 24)),
511+
input_size=(3, 384, 384), pool_size=(24, 24), crop_mode='squash', crop_pct=1.0),
512512

513513
'cvt_21.msft_in1k': _cfg(url='https://github.com/fffffgggg54/pytorch-image-models/releases/download/cvt/CvT-21-224x224-IN-1k.pth'),
514514
'cvt_21.msft_in1k_384': _cfg(url='https://github.com/fffffgggg54/pytorch-image-models/releases/download/cvt/CvT-21-384x384-IN-1k.pth',
515-
input_size=(3, 384, 384), pool_size=(24, 24)),
515+
input_size=(3, 384, 384), pool_size=(24, 24), crop_mode='squash', crop_pct=1.0),
516516
'cvt_21.msft_in22k_ft_in1k_384': _cfg(url='https://github.com/fffffgggg54/pytorch-image-models/releases/download/cvt/CvT-21-384x384-IN-22k.pth',
517-
input_size=(3, 384, 384), pool_size=(24, 24)),
517+
input_size=(3, 384, 384), pool_size=(24, 24), crop_mode='squash', crop_pct=1.0),
518518

519519
'cvt_w24.msft_in22k_ft_in1k_384': _cfg(url='https://github.com/fffffgggg54/pytorch-image-models/releases/download/cvt/CvT-w24-384x384-IN-22k.pth',
520-
input_size=(3, 384, 384), pool_size=(24, 24)),
520+
input_size=(3, 384, 384), pool_size=(24, 24), crop_mode='squash', crop_pct=1.0),
521521
})
522522

523523

0 commit comments

Comments
 (0)