Skip to content

PyTorch implementaion of Data-eifficient Image Transformers. I have written code for student network and teacher network sperately and then combined them together.

Notifications You must be signed in to change notification settings

sourabh-patil/DeiT-Training-data-efficient-image-transformers-distillation-through-attention

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 

Repository files navigation

DeiT-Training-data-efficient-image-transformers-distillation-through-attention

PyTorch implementaion of Data-eifficient Image Transformers. I have written code for student network and teacher network sperately and then combined them together. Like to the paper: https://arxiv.org/pdf/2012.12877.pdf The student network is a vision transformer which learns itself as well as takes valueable lessons from the teacher network. Teacher network is a convolutional network which is pre-trained. For time being I have used Teacher network which is custom built. One can use different variants of ResNet, VGG, etc. according to the use case. Following is the overall block diagram shown in the paper.

deit_block

About

PyTorch implementaion of Data-eifficient Image Transformers. I have written code for student network and teacher network sperately and then combined them together.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages