code: def __init__(self):
      super().__init__()
      self.linear1 = nn.Linear(4, 4)
      self.relu = nn.ReLU(inplace=True)
      self.linear2 = nn.Linear(4, 4)
  
file_path: ColossalAI/tests/test_auto_parallel/test_tensor_shard/test_liveness_analysis.py
imports: import pytest
  import torch
  import torch.nn
  from colossalai._analyzer.fx.graph_module import ColoGraphModule
  from colossalai._analyzer.fx.passes import shape_prop_pass
  from colossalai._analyzer.fx.tracer.tracer import ColoTracer
  from colossalai.auto_parallel.tensor_shard.solver import GraphAnalyser
  from colossalai.fx import ColoGraphModule, ColoTracer
  from colossalai.testing import clear_cache_before_run
label_desc: Use image node_id 1 to create a new LinearModel object from inherited base classes: nn with example: obj = LinearModel()