search

Sense-X / Co-DETR

[ICCV 2023] DETRs with Collaborative Hybrid Assignments Training
MIT License
950 stars 100 forks source link

frozen ViT-L backbone #126

Closed hubhub086 closed 4 months ago

hubhub086 commented 5 months ago

hello! I want to freeze the backbone of Co-DETR-ViT-L，so how can I do it? Is there any method like the frozen_stages in Resnet backbone?

hubhub086 commented 5 months ago

can I simply add this code? Snipaste_2024-03-27_13-40-01

TempleX98 commented 4 months ago

Freezing the backbone has been implemented in _freeze_stages of mmdet/models/backbones/swin.py.

hubhub086 commented 4 months ago

Thanks!