code: def initialize_norm_layers(module) -> None:
      """Helper function to initialize normalization layers"""
  
      def _init_fn(module):
          if hasattr(module, "weight") and module.weight is not None:
              nn.init.ones_(module.weight)
          if hasattr(module, "bias") and module.bias is not None:
              nn.init.zeros_(module.bias)
  
      _init_fn(module.layer) if hasattr(module, "layer") else _init_fn(
          module=module
      )
  
file_path: ml-cvnets/cvnets/misc/init_utils.py
imports: from typing import Optional
  from torch import nn
  from cvnets.layers import GroupLinear, LinearLayer, norm_layers_tuple
  from utils import logger
label_desc: Use image node_id 4 for calling a global function with example usage: initialize_norm_layers(module) without return types