code: def __init__(self, checkpoint=False) -> None:
      super().__init__(checkpoint)
      self.fc1 = nn.Linear(5, 5)
      self.sub_fc = SubNet(5)
      self.fc2 = nn.Linear(5, 2)
  
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 NestedNet object from inherited base classes: CheckpointModule with example: obj = NestedNet(checkpoint)