code: def init_params(net):
      """Init layer parameters."""
      for module in net.modules():
          if isinstance(module, nn.Conv2d):
              init.kaiming_normal(module.weight, mode="fan_out")
              if module.bias:
                  init.constant(module.bias, 0)
          elif isinstance(module, nn.BatchNorm2d):
              init.constant(module.weight, 1)
              init.constant(module.bias, 0)
          elif isinstance(module, nn.Linear):
              init.normal(module.weight, std=1e-3)
              if module.bias:
                  init.constant(module.bias, 0)
  
file_path: nni/examples/trials/network_morphism/cifar10/utils.py
imports: import numpy
  import torch
  import torch.nn
  import torch.nn.init
  import torchvision.transforms
label_desc: Use image node_id 4 for calling a global function with example usage: init_params(net) without return types