code: def __init__(
      self,
      module: torch.nn.Module,
      module_name: str,
      module_full_name: str,
      parent_module: torch.nn.Module,
  ):
      self.module = module
      self.module_name = module_name
      self.module_full_name = module_full_name
      self.parent_module = parent_module
  
file_path: colossalai/colossalai/zero/gemini/memory_tracer/static_memstats_collector.py
imports: from typing import Optional
  import torch
  import torch.nn
  from torch.fx import symbolic_trace
  from colossalai.fx.passes.meta_info_prop import MetaInfoProp
  from colossalai.fx.profiler import calculate_fwd_out, calculate_fwd_tmp, is_compatible_with_meta
  from colossalai.zero.gemini.chunk import ChunkManager
  from .chunk_memstats_collector import ChunkMemStatsCollector
label_desc: Use image node_id 1 to create a new ModuleInfos object  with example: obj = ModuleInfos(module, module_name, module_full_name, parent_module)