timm

/

convnext_base.fb_in22k_ft_in1k

like 2

Follow

PyTorch Image Models 183

I want to use convnext v2 . So i tried the the feature map script : It gave me following error.

RuntimeError Traceback (most recent call last)
Cell In[1], line 9
3 import timm
5 img = Image.open(urlopen(
6 'https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/beignets-task-guide.png'
7 ))
----> 9 model = timm.create_model(
10 'convnext_base.fb_in22k_ft_in1k',
11 pretrained=True,
12 features_only=True,
13 )
14 model = model.eval()
16 # get model specific transforms (normalization, resize)

File ~/venv/parenv/lib/python3.10/site-packages/timm/models/factory.py:67, in create_model(model_name, pretrained, pretrained_cfg, checkpoint_path, scriptable, exportable, no_jit, **kwargs)
64 pretrained_cfg, model_name = load_model_config_from_hf(model_name)
66 if not is_model(model_name):
---> 67 raise RuntimeError('Unknown model (%s)' % model_name)
69 create_fn = model_entrypoint(model_name)
70 with set_layer_config(scriptable=scriptable, exportable=exportable, no_jit=no_jit):

RuntimeError: Unknown model (convnext_base.fb_in22k_ft_in1k)