Selective Layer Distillation for Vision Transformers
- Semin Kim
- Jeonghwan Cho
- Seonah Ryu
- Xu Meow
- Clone this repo and run the commands below
git clone https://github.com/serizard/DeltaKD.git && cd AAAKD
- Install Packages
conda create -n deltakd python=3.10 -y
conda activate deltakd
pip install -e .
cd AAAKD
bash exp/baseline-deit-tiny.sh $GPU_IDS
GPU_IDS -> ex: 0,1,2,3 (GPU ids to use)
Feel free to customize other arguments and configs.