@@ -805,36 +805,35 @@ def checkpoint_filter_fn(
805805 return state_dict
806806
807807
808+ def _cfg (url = '' , ** kwargs ):
809+ return {
810+ 'license' : 'apache-2.0' ,
811+ 'num_classes' : 0 ,
812+ 'interpolation' : 'bilinear' ,
813+ 'fixed_input_size' : True ,
814+ 'mean' : IMAGENET_INCEPTION_MEAN ,
815+ 'std' : IMAGENET_INCEPTION_STD ,
816+ ** kwargs
817+ }
818+
808819default_cfgs = generate_default_cfgs ({
809820 'pe_core_b16_224' : _cfg (
810821 hf_hub_id = 'timm/' ,
811- license = 'apache-2.0' ,
812- mean = IMAGENET_INCEPTION_MEAN , std = IMAGENET_INCEPTION_STD , num_classes = 0 ,
813822 input_size = (3 , 224 , 224 )),
814823 'pe_core_l14_336' : _cfg (
815824 hf_hub_id = 'timm/' ,
816- license = 'apache-2.0' ,
817- mean = IMAGENET_INCEPTION_MEAN , std = IMAGENET_INCEPTION_STD , num_classes = 0 ,
818825 input_size = (3 , 336 , 336 )),
819826 'pe_core_G14_448' : _cfg (
820827 hf_hub_id = 'timm/' ,
821- license = 'apache-2.0' ,
822- mean = IMAGENET_INCEPTION_MEAN , std = IMAGENET_INCEPTION_STD , num_classes = 0 ,
823828 input_size = (3 , 448 , 448 )),
824829 'pe_lang_l14_448' : _cfg (
825830 hf_hub_id = 'timm/' ,
826- license = 'apache-2.0' ,
827- mean = IMAGENET_INCEPTION_MEAN , std = IMAGENET_INCEPTION_STD , num_classes = 0 ,
828831 input_size = (3 , 448 , 448 )),
829832 'pe_lang_G14_448' : _cfg (
830833 hf_hub_id = 'timm/' ,
831- license = 'apache-2.0' ,
832- mean = IMAGENET_INCEPTION_MEAN , std = IMAGENET_INCEPTION_STD , num_classes = 0 ,
833834 input_size = (3 , 448 , 448 )),
834835 'pe_spatial_G14_448' : _cfg (
835836 hf_hub_id = 'timm/' ,
836- license = 'apache-2.0' ,
837- mean = IMAGENET_INCEPTION_MEAN , std = IMAGENET_INCEPTION_STD , num_classes = 0 ,
838837 input_size = (3 , 448 , 448 )),
839838})
840839
0 commit comments