code: def initialize_weights(model):
      """
      Function to initialize network weights.
      NOTE: NOT USED IN MAIN SCRIPT.
  
      Args:
          model: PyTorch Network
      Returns:
          Nothing!
      """
      for idx, module in enumerate(model.modules()):
          if isinstance(module, nn.Conv2d):
              nn.init.kaiming_normal_(
                  module.weight, mode="fan_out", nonlinearity="relu"
              )
          elif isinstance(module, nn.BatchNorm2d):
              nn.init.constant_(module.weight, 1)
              nn.init.constant_(module.bias, 0)
          elif isinstance(module, nn.Linear):
              module.weight.data.normal_(0, 0.01)
              module.bias.data.zero_()
  
file_path: dgl/examples/pytorch/hilander/PSS/Smooth_AP/src/netlib.py
imports: import os
  from collections import OrderedDict
  import auxiliaries
  import pretrainedmodels
  import torch
  import torch.nn
label_desc: Use image node_id 1 for calling a global function with example usage: initialize_weights(model) without return types