search

Sense-X / Co-DETR

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

freeze most of the layers and train head #109

Open Qia98 opened 7 months ago

Qia98 commented 7 months ago

I would like to train on a custom dataset, but I want to freeze most of the layers and only train the last few layers (like head or decoder). I hope that the mAP or AP metric can reach the results of training the YOLO model on the same dataset. Is this feasible?

TempleX98 commented 6 months ago

I have yet to explore this configuration. Perhaps you can perform this experiment using your custom dataset to benchmark this setting.