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.
-
Notifications
You must be signed in to change notification settings - Fork 1
sourabh-patil/DeiT-Training-data-efficient-image-transformers-distillation-through-attention
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
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 0
No packages published