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
| Name | Name | Last commit date | ||
|---|---|---|---|---|
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