fix init of convfchead (#6624)

open-mmlab · Dec 14, 2021 · 70ecbd8 · 70ecbd8
1 parent 1260f29
commit 70ecbd8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mmdet/models/roi_heads/bbox_heads/convfc_bbox_head.py b/mmdet/models/roi_heads/bbox_heads/convfc_bbox_head.py
@@ -107,6 +107,7 @@ def __init__(self,
             self.init_cfg += [
                 dict(
                     type='Xavier',
+                    distribution='uniform',
                     override=[
                         dict(name='shared_fcs'),
                         dict(name='cls_fcs'),