code: def _init_vit_weights(m):
      """
      ViT weight initialization
      :param m: module
      """
      if isinstance(m, nn.Linear):
          nn.init.trunc_normal_(m.weight, std=0.01)
          if m.bias is not None:
              nn.init.zeros_(m.bias)
      elif isinstance(m, nn.Conv2d):
          nn.init.kaiming_normal_(m.weight, mode="fan_out")
          if m.bias is not None:
              nn.init.zeros_(m.bias)
      elif isinstance(m, nn.LayerNorm):
          nn.init.zeros_(m.bias)
          nn.init.ones_(m.weight)
  
imports: from functools import partial
  from collections import OrderedDict
  import torch
  import torch.nn
label_desc: Use image node_id 2 for calling a global function with example usage: _init_vit_weights(m) without return types