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

Selective QR #4

Open
ykyk000 opened this issue Mar 6, 2023 · 1 comment
Open

Selective QR #4

ykyk000 opened this issue Mar 6, 2023 · 1 comment

Comments

@ykyk000
Copy link

ykyk000 commented Mar 6, 2023

Thanks for your work. when reading your article and the published code, I would like to ask which files apply the Selective Query Recollection method in the published code.

@Fangyi-Chen
Copy link
Owner

Fangyi-Chen commented Mar 7, 2023

Hi,

please refer to the Readme.md “guide to code". Thanks.

SQR-Adamixer
We provide two implementation instances of SQR-adamixer in this repo, one is in /mmdet/models/roi_heads/adamixer_decoder_Qrecycle.py, which might be slower for training but require less GPU memory (and easy to understand the logic). Another is in /mmdet/models/roi_heads/adamixer_decoder_Qrecycle_optimize.py, which is much faster than the former (and recommended for using) but has higher requirement on GPU memory.

SQR-Deformable DETR
Similarly, We provide two implementation instances of SQR-deformable DETR in QRDeformableDetrTransformerDecoder in /mmdet/models/utils/transformer.py. Named as forward and forward_slow, separately.

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