code: def initialize_weights(module):
      if isinstance(module, nn.Conv2d):
          nn.init.kaiming_normal_(module.weight.data, mode="fan_out")
      elif isinstance(module, nn.BatchNorm2d):
          module.weight.data.fill_(1)
          module.bias.data.zero_()
      elif isinstance(module, nn.Linear):
          module.bias.data.zero_()
  
imports: import torch
  import torch.nn
  import torch.nn.functional
  from collections import OrderedDict
label_desc: Use image node_id 3 for calling a global function with example usage: initialize_weights(module) without return types