Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[FEATURE_REQUEST] Add vision transformers model to image classification #678

Open
kbjiang opened this issue Oct 27, 2022 · 1 comment
Open

Comments

@kbjiang
Copy link
Contributor

kbjiang commented Oct 27, 2022

Description

The transformer-based image classification model is becoming popular. It will be nice to include it in this repo.

Expected behavior with the suggested feature

  • similar classification performance as the CNN based models

Other Comments

@ShreyasJoshi7
Copy link

ShreyasJoshi7 commented Feb 17, 2024

@kbjiang I have created a PR which demonstrates the inference of transformer-based image classification using DeiT. I would appreciate if you could have a look at it. Let me know if you need more things to be added to it. This is the link to the PR
Cheers!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants