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_()
  
file_path: pywick/pywick/models/segmentation/gcnnets/gcn.py
imports: import torch
  import torch.nn.functional
  from torch import nn
  from torchvision import models
  import os
  from math import floor
label_desc: Use image node_id 1 for calling a global function with example usage: initialize_weights() without return types