From 3af564f2fb3727cac106a20bb9124f8f75d3f306 Mon Sep 17 00:00:00 2001 From: berniebear Date: Fri, 25 Apr 2025 09:19:18 +0000 Subject: [PATCH] fix default config --- timm/models/pe.py | 23 +++++++++++------------ 1 file changed, 11 insertions(+), 12 deletions(-) diff --git a/timm/models/pe.py b/timm/models/pe.py index 717215c1..953a9906 100644 --- a/timm/models/pe.py +++ b/timm/models/pe.py @@ -805,36 +805,35 @@ def checkpoint_filter_fn( return state_dict +def _cfg(url='', **kwargs): + return { + 'license': 'apache-2.0', + 'num_classes': 0, + 'interpolation': 'bilinear', + 'fixed_input_size': True, + 'mean': IMAGENET_INCEPTION_MEAN, + 'std': IMAGENET_INCEPTION_STD, + **kwargs + } + default_cfgs = generate_default_cfgs({ 'pe_core_b16_224': _cfg( hf_hub_id='timm/', - license='apache-2.0', - mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, num_classes=0, input_size=(3, 224, 224)), 'pe_core_l14_336': _cfg( hf_hub_id='timm/', - license='apache-2.0', - mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, num_classes=0, input_size=(3, 336, 336)), 'pe_core_G14_448': _cfg( hf_hub_id='timm/', - license='apache-2.0', - mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, num_classes=0, input_size=(3, 448, 448)), 'pe_lang_l14_448': _cfg( hf_hub_id='timm/', - license='apache-2.0', - mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, num_classes=0, input_size=(3, 448, 448)), 'pe_lang_G14_448': _cfg( hf_hub_id='timm/', - license='apache-2.0', - mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, num_classes=0, input_size=(3, 448, 448)), 'pe_spatial_G14_448': _cfg( hf_hub_id='timm/', - license='apache-2.0', - mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, num_classes=0, input_size=(3, 448, 448)), })