diff --git a/configs/vitdet/cascade_rcnn_vit_base_hrfpn_cae_1x_coco.yml b/configs/vitdet/cascade_rcnn_vit_base_hrfpn_cae_1x_coco.yml index 9ffbab297d14439712b105bd1b855315da467714..f7808b7cc3ed90be63a984f8552731e3e0e289f7 100644 --- a/configs/vitdet/cascade_rcnn_vit_base_hrfpn_cae_1x_coco.yml +++ b/configs/vitdet/cascade_rcnn_vit_base_hrfpn_cae_1x_coco.yml @@ -98,6 +98,7 @@ CascadeHead: reg_class_agnostic: False stage_loss_weights: [1, 0.5, 0.25] loss_normalize_pos: True + add_gt_as_proposals: [True, True, True] BBoxAssigner: