@@ -495,9 +495,9 @@ def _cfg(url='', **kwargs):
495
495
return {
496
496
'url' : url ,
497
497
'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' ,
499
499
'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' ,
501
501
** kwargs
502
502
}
503
503
@@ -506,18 +506,18 @@ def _cfg(url='', **kwargs):
506
506
url = 'https://github.com/fffffgggg54/pytorch-image-models/releases/download/cvt/CvT-13-224x224-IN-1k.pth' ),
507
507
'cvt_13.msft_in1k_384' : _cfg (
508
508
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 ),
510
510
'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 ),
512
512
513
513
'cvt_21.msft_in1k' : _cfg (url = 'https://github.com/fffffgggg54/pytorch-image-models/releases/download/cvt/CvT-21-224x224-IN-1k.pth' ),
514
514
'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 ),
516
516
'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 ),
518
518
519
519
'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 ),
521
521
})
522
522
523
523
0 commit comments