def METHOD_NAME(self, **kwargs):
    """Weight Initialization for ACRNHead."""
    for m in self.modules():
        if isinstance(m, nn.Conv3d):
            kaiming_init(m)
        elif isinstance(m, _BatchNorm):
            constant_init(m, 1)