YOLOF icon indicating copy to clipboard operation
YOLOF copied to clipboard

Freeze the Backbone and Dilated Encoder

Open ukdsvl opened this issue 3 years ago • 1 comments

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 ?

ukdsvl avatar Sep 10 '21 14:09 ukdsvl

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

chensnathan avatar Sep 11 '21 10:09 chensnathan