code: def _init_weights(self, m):
      if isinstance(m, nn.Linear):
          if self.weight_init_style == "jax":
              # Based on MAE and official Jax ViT implementation
              torch.nn.init.xavier_uniform_(m.weight)
          elif self.weight_init_style == "pytorch":
              # PyTorch ViT uses trunc_normal_
              trunc_normal_(m.weight, std=0.02)
  
          if m.bias is not None:
              nn.init.constant_(m.bias, 0)
      elif isinstance(m, (nn.LayerNorm)):
          nn.init.constant_(m.bias, 0)
          nn.init.constant_(m.weight, 1.0)
  
imports: from functools import partial
  from typing import Callable, List, Optional
  import torch
  import torch.nn
  import torch.utils.checkpoint
  from timm.models.layers import DropPath, trunc_normal_
label_desc: Use image node_id 2 for calling the SimpleTransformer obj's underlying member method code with example usage: obj._init_weights(m) without return types