mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

fix default config

Browse Source
This commit is contained in:
berniebear 2025-04-25 09:19:18 +00:00
parent e6bbf9fd56
commit 3af564f2fb
1 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
timm/models/pe.py
View File

@ -805,36 +805,35 @@ def checkpoint_filter_fn(
return state_dict 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({ default_cfgs = generate_default_cfgs({
'pe_core_b16_224': _cfg( 'pe_core_b16_224': _cfg(
hf_hub_id='timm/', hf_hub_id='timm/',
license='apache-2.0',
mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, num_classes=0,
input_size=(3, 224, 224)), input_size=(3, 224, 224)),
'pe_core_l14_336': _cfg( 'pe_core_l14_336': _cfg(
hf_hub_id='timm/', hf_hub_id='timm/',
license='apache-2.0',
mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, num_classes=0,
input_size=(3, 336, 336)), input_size=(3, 336, 336)),
'pe_core_G14_448': _cfg( 'pe_core_G14_448': _cfg(
hf_hub_id='timm/', hf_hub_id='timm/',
license='apache-2.0',
mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, num_classes=0,
input_size=(3, 448, 448)), input_size=(3, 448, 448)),
'pe_lang_l14_448': _cfg( 'pe_lang_l14_448': _cfg(
hf_hub_id='timm/', hf_hub_id='timm/',
license='apache-2.0',
mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, num_classes=0,
input_size=(3, 448, 448)), input_size=(3, 448, 448)),
'pe_lang_G14_448': _cfg( 'pe_lang_G14_448': _cfg(
hf_hub_id='timm/', hf_hub_id='timm/',
license='apache-2.0',
mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, num_classes=0,
input_size=(3, 448, 448)), input_size=(3, 448, 448)),
'pe_spatial_G14_448': _cfg( 'pe_spatial_G14_448': _cfg(
hf_hub_id='timm/', hf_hub_id='timm/',
license='apache-2.0',
mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, num_classes=0,
input_size=(3, 448, 448)), input_size=(3, 448, 448)),
}) })