code: def test_init_weights(self):
      head = YOLOXHead(
          num_classes=4,
          in_channels=1,
          stacked_convs=1,
          use_depthwise=False,
      )
      head.init_weights()
      bias_init = bias_init_with_prob(0.01)
      for conv_cls, conv_obj in zip(
          head.multi_level_conv_cls, head.multi_level_conv_obj
      ):
          assert_allclose(
              conv_cls.bias.data,
              torch.ones_like(conv_cls.bias.data) * bias_init,
          )
          assert_allclose(
              conv_obj.bias.data,
              torch.ones_like(conv_obj.bias.data) * bias_init,
          )
  
file_path: mmdetection/tests/test_models/test_dense_heads/test_yolox_head.py
imports: from unittest import TestCase
  import torch
  from mmcv.cnn import ConvModule, DepthwiseSeparableConvModule
  from mmengine.config import Config
  from mmengine.model import bias_init_with_prob
  from mmengine.structures import InstanceData
  from mmengine.testing import assert_allclose
  from mmdet.models.dense_heads import YOLOXHead
label_desc: Use image node_id 1 for calling the TestYOLOXHead obj's underlying member method code with example usage: obj.test_init_weights() without return types