code: def __init__(self):
      super().__init__()
      self.a = nn.Linear(10, 10)
      self.b = nn.Linear(10, 10)
      self.c = nn.Linear(10, 10)
      self.d = nn.Linear(10, 10)
      self.e = nn.Linear(10, 10)
  
file_path: ColossalAI/tests/test_analyzer/test_fx/test_nested_ckpt.py
imports: import pytest
  import torch
  import torch.nn
  from torch.utils.checkpoint import checkpoint
  from colossalai.testing import clear_cache_before_run
label_desc: Use image node_id 1 to create a new MyModule object from inherited base classes: nn with example: obj = MyModule()