To include other vision transformer models - Githubissues

rosinality / vision-transformers-pytorch

Implementation of various Vision Transformers I found interesting

MIT License

84 stars 11 forks source link

To include other vision transformer models #1

Open rishikksh20 opened 3 years ago

rishikksh20 commented 3 years ago

Hi @rosinality, hope you are doing well! I really like your repo, especially for dataloader and augmentation part for image classification. I am not majorly working on Vision field but still I have implemented some of the vision transformer models, following:

https://github.com/rishikksh20/convolution-vision-transformers
https://github.com/rishikksh20/CeiT-pytorch
https://github.com/rishikksh20/CrossViT-pytorch
https://github.com/rishikksh20/ViViT-pytorch And happy to include all these into this repo.

rosinality commented 3 years ago

Thank you, looks very interesting! I will check it.