Cannot import vit_deit_small

SwinTransformer / Transformer-SSL

This is an official implementation for "Self-Supervised Learning with Swin Transformers".

https://arxiv.org/abs/2105.04553

MIT License

621 stars 66 forks source link

Cannot import vit_deit_small #3

Open michuanhaohao opened 3 years ago

michuanhaohao commented 3 years ago

from timm.models import vit_deit_small_patch16_224 ImportError: cannot import name 'vit_deit_small_patch16_224' from 'timm.models' (/home/michuan.lh/miniconda3/envs/moby/lib/python3.7/site-packages/timm/models/init.py)

Thanks for your work. When I run your code, I got an error that cannot import vit_deit_small_patch16_224.

launchauto commented 3 years ago

pip install timm==0.4.9 or update to the newest version of timm package would help