search

Fangyi-Chen / SQR

MIT License
102 stars 5 forks source link

Selective QR #4

Open ykyk000 opened 1 year ago

ykyk000 commented 1 year ago

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 commented 1 year ago

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.