facebookresearch / LeViT

LeViT a Vision Transformer in ConvNet's Clothing for Faster Inference

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

AttributeError: module 'timm.models' has no attribute 'vit_deit_tiny_distilled_patch16_224'

songtianhui opened this issue · comments

Hello!
When I run speed_test.py, error AttributeError: module 'timm.models' has no attribute 'vit_deit_tiny_distilled_patch16_224'.
Does it mean that my timm doesn't have the deit model? How can I solve it?
I go to timm repo https://github.com/rwightman/pytorch-image-models but didn't find how to add models.
Thanks!

timm changed the model names it seems. Now updated. Thank you for reporting.