Implement of Adaptive Sparse ViT (IJCAI23)
torch==1.9.0
torchvision==0.10.1
timm==0.4.12
sh finetune.sh
python3 infer.py --model-path path_to_model.pth --batch-size 256 --arch deit_small --data-path path_to_imagenet
-
Notifications
You must be signed in to change notification settings - Fork 1
Cydia2018/AS-ViT
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Adaptive Sparse ViT
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published