timm
/

Image Classification
timm
PyTorch
Safetensors

Is there an issue when using this model in timm??

#1
by X01D - opened

when trying to use the model by timm
using this code

import timm
model = timm.create_model("hf_hub:timm/mobilenetv4_conv_small.e1200_r224_in1k", pretrained=True)

produces this error
"RuntimeError: Unknown model (mobilenetv4_conv_small)"
how could I use it?

PyTorch Image Models org
edited Jun 14

@X01D you need to be using the timm main branch right now, it's not in a pypi pip installable release yet... within a few days.

pip install git+https://github.com/rwightman/pytorch-image-models.git

should be 1.0.6dev0 version if installed correctly

Thank you sir

X01D changed discussion status to closed

Sign up or log in to comment