Input size of an image

[bekhzod-olimov]

How to automatically get the input size of an image after loading a model using create_model from timm.list_models list using for loop?

for i, model_name in enumerate(list(timm.list_models())):
    model = timm.create_model(model_name)
    inp_size = model...?

[rwightman]

@bekhzod-olimov

>>> m = timm.create_model('vit_base_patch16_224')
>>> pprint(m.pretrained_cfg)
{'architecture': 'vit_base_patch16_224',
 'classifier': 'head',
 'crop_mode': 'center',
 'crop_pct': 0.9,
 'custom_load': False,
 'first_conv': 'patch_embed.proj',
 'fixed_input_size': True,
 'hf_hub_id': 'timm/vit_base_patch16_224.augreg2_in21k_ft_in1k',
 'input_size': (3, 224, 224),
 'interpolation': 'bicubic',
 'mean': (0.5, 0.5, 0.5),
 'num_classes': 1000,
 'pool_size': None,
 'std': (0.5, 0.5, 0.5),
 'tag': 'augreg2_in21k_ft_in1k',
 'url': ''}

[bekhzod-olimov]

@rwightman Thank you! I got what I was looking for!