code: def __init__(self) -> None:
      super().__init__()
      self.layers = nn.Sequential(nn.Linear(1, 2), nn.Linear(2, 3))
      self.out = nn.Linear(3, 1)
  
imports: import torch
  import torch.nn
  from colossalai.shardformer.shard.utils import set_tensors_to_none
label_desc: Use image node_id 1 to create a new Net object from inherited base classes: nn with example: obj = Net()