Skip to content

Selective Layer Distillation for Vision Transformers

License

Notifications You must be signed in to change notification settings

serizard/DeltaKD

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

53 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DeltaKD (1st brAIn research program)

Selective Layer Distillation for Vision Transformers

Contributors:

  • Semin Kim
  • Jeonghwan Cho
  • Seonah Ryu
  • Xu Meow

Install

  1. Clone this repo and run the commands below
git clone https://github.com/serizard/DeltaKD.git && cd AAAKD 
  1. Install Packages
conda create -n deltakd python=3.10 -y
conda activate deltakd
pip install -e .

Run experiments

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.

About

Selective Layer Distillation for Vision Transformers

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 3

  •  
  •  
  •