Co-DETR icon indicating copy to clipboard operation
Co-DETR copied to clipboard

freeze most of the layers and train head

Open Qia98 opened this issue 1 year ago • 1 comments

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?

Qia98 avatar Jan 30 '24 08:01 Qia98

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

TempleX98 avatar Feb 01 '24 09:02 TempleX98