code: def __init__(self, out_features) -> None:
      super().__init__()
      self.bias = nn.Parameter(torch.zeros(out_features))
  
file_path: ColossalAI/tests/kit/model_zoo/custom/nested_model.py
imports: import torch
  import torch.nn
  import torch.nn.functional
  from ..registry import model_zoo
  from .base import CheckpointModule
label_desc: Use image node_id 1 to create a new SubNet object from inherited base classes: nn with example: obj = SubNet(out_features)