docstr: Memory Statistic Collector for Chunks.
  
  Args:
      chunk_manager (ChunkManager): the chunk manager.
      memstats (Optional[MemStats], optional): memory statistics collected by RMT. Defaults to None.
code: def __init__(
      self,
      chunk_manager: ChunkManager,
      memstats: Optional[MemStats] = None,
  ) -> None:
      """
  
      Memory Statistic Collector for Chunks.
  
      Args:
          chunk_manager (ChunkManager): the chunk manager.
          memstats (Optional[MemStats], optional): memory statistics collected by RMT. Defaults to None.
      """
      super().__init__(memstats)
      self._chunk_manager = chunk_manager
  
file_path: colossalai/colossalai/zero/gemini/memory_tracer/chunk_memstats_collector.py
imports: from typing import Optional
  from colossalai.utils import get_current_device
  from colossalai.zero.gemini.chunk import ChunkManager
  from .memory_stats import MemStats
  from .memstats_collector import MemStatsCollector
label_desc: Use image node_id 1 to create a new ChunkMemStatsCollector object from inherited base classes: MemStatsCollector with example: obj = ChunkMemStatsCollector(chunk_manager, memstats)