RuntimeError: Unknown model (mobilevit)

huggingface / pytorch-image-models

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

https://huggingface.co/docs/timm

Apache License 2.0

31.94k stars 4.73k forks source link

RuntimeError: Unknown model (mobilevit) #1215

Closed MrWcy closed 2 years ago

MrWcy commented 2 years ago

When I want to use mobilevit, it throws an error

rwightman commented 2 years ago

@MrWcy there is no model with that name

https://github.com/rwightman/pytorch-image-models/blob/master/timm/models/mobilevit.py#L255-L268

you also need to be using the current master (pip install git+https://github.com/rwightman/pytorch-image-models) version 0.6.x not pypi pip install timm 0.5.x release.