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.

Sign up for GitHub

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

Jump to bottom

Hope the whole source code can be released #1

Open
qgh1223 opened this issue Apr 2, 2023 · 3 comments
Open

Hope the whole source code can be released #1

qgh1223 opened this issue Apr 2, 2023 · 3 comments

Comments

@qgh1223
Copy link

qgh1223 commented Apr 2, 2023

Hope the whole source code can be released
@SiyuanWei
Copy link
Collaborator

Thanks ! We are working on this !

@LiuDongyang6
Copy link

Hi! just want to follow up on this to see if there is any update. Would you release the eTPS codes? Thanks!

Thanks ! We are working on this !

@SiyuanWei
Copy link
Collaborator

Sorry for the late reply. The codes of our eTPS implemented by megengine can be found in https://github.com/megvii-research/TPS-CVPR2023/blob/main/megengine_codebase/models/tps.py#L91. Sorry that we haven't released the whole implementation of eTPS on the vit block. It is injected after multi-head attention block and utilizes the class token score to group tokens. If you have more problems, you can reach me in email.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@SiyuanWei @qgh1223 @LiuDongyang6