AS-ViT

Implement of Adaptive Sparse ViT (IJCAI23)

requirement

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

Name		Name	Last commit message	Last commit date
Latest commit History 3 Commits
figs		figs
README.md		README.md
datasets.py		datasets.py
engine.py		engine.py
finetune.sh		finetune.sh
helpers.py		helpers.py
infer.py		infer.py
losses.py		losses.py
lvvit.py		lvvit.py
main.py		main.py
samplers.py		samplers.py
utils.py		utils.py
vit.py		vit.py
viz.py		viz.py