code: def initialize_weights(*models):
      for model in models:
          for module in model.modules():
              if isinstance(module, (nn.Conv2d, nn.Linear)):
                  nn.init.kaiming_normal_(module.weight)
                  if module.bias is not None:
                      module.bias.data.zero_()
              elif isinstance(module, nn.BatchNorm2d):
                  module.weight.data.fill_(1)
                  module.bias.data.zero_()
  
imports: import os
  from math import ceil
  import numpy
  import torch
  import torch.nn.functional
  from torch import nn
label_desc: Use image node_id 2 for calling a global function with example usage: initialize_weights() without return types