Skip to content

Commit 3af564f

Browse files
committed
fix default config
1 parent e6bbf9f commit 3af564f

File tree

1 file changed

+11
-12
lines changed

1 file changed

+11
-12
lines changed

timm/models/pe.py

Lines changed: 11 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -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+
808819
default_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

Comments
 (0)