search

chensnathan / YOLOF

You Only Look One-level Feature (YOLOF), CVPR2021, Detectron2
MIT License
271 stars 28 forks source link

Freeze the Backbone and Dilated Encoder #30

Open ukdsvl opened 3 years ago

ukdsvl commented 3 years ago

Hi,

Many thanks for your awesome work. In the original Detectron2, we can freeze certain layers of the backbone with _C.MODEL.BACKBONE.FREEZE_AT.

Can you please let me know how to freeze both the ResNet backbone AND the Dilated Encoder, and update only the Decoder during training in YOLOF as well, say using yolof_R_50_C5_1x ?

chensnathan commented 3 years ago

You can manually add the freeze codes for backbone and encoder here in the yolof file as we do in the darknet.