@@ -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