class_docstr: GPT Transformer Model
class_docstr_tok: ["GPT","Transformer","Model"]
code: def _init_weights(self, module):
      if isinstance(module, nn.Linear):
          torch.nn.init.normal_(module.weight, mean=0.0, std=0.02)
          if module.bias is not None:
              torch.nn.init.zeros_(module.bias)
      elif isinstance(module, nn.Embedding):
          torch.nn.init.normal_(module.weight, mean=0.0, std=0.02)
      elif isinstance(module, nn.LayerNorm):
          torch.nn.init.zeros_(module.bias)
          torch.nn.init.ones_(module.weight)
  
imports: import math
  from dataclasses import dataclass
  from typing import Tuple
  import torch
  import torch.nn
  from torch.nn import functional
  from ray.rllib.utils.annotations import DeveloperAPI
  from ray.rllib.utils.deprecation import Deprecated
label_desc: Use image node_id 2 for calling the GPT obj's underlying member method code with example usage: obj._init_weights(module) without return types