def init_weights(self):
    super(ConvFCBBoxHead, self).init_weights()
    if self.with_dis:
        for module_list in [self.shared_fcs, self.cls_fcs, self.reg_fcs, self.dis_fcs]:
            for m in module_list.modules():
                if isinstance(m, nn.Linear):
                    nn.init.xavier_uniform_(m.weight)
                    nn.init.constant_(m.bias, 0)
    else:
        for module_list in [self.shared_fcs, self.cls_fcs, self.reg_fcs]:
            for m in module_list.modules():
                if isinstance(m, nn.Linear):
                    nn.init.xavier_uniform_(m.weight)
                    nn.init.constant_(m.bias, 0)
def init_weights(self):
    super(ConvFCBBoxHead, self).init_weights()
    if self.with_dis:
        for module_list in [self.shared_fcs, self.cls_fcs, self.reg_fcs, self.dis_fcs]:
            for m in module_list.modules():
                if isinstance(m, nn.Linear):
                    nn.init.xavier_uniform_(m.weight)
                    nn.init.constant_(m.bias, 0)
    else:
        for module_list in [self.shared_fcs, self.cls_fcs, self.reg_fcs]:
            for m in module_list.modules():
                if isinstance(m, nn.Linear):
                    nn.init.xavier_uniform_(m.weight)
                    nn.init.constant_(m.bias, 0)