code: def cuda_margin_mem(self) -> float:
      from colossalai.legacy.utils.memory import (
          colo_device_memory_capacity,
      )
  
      return (
          colo_device_memory_capacity(get_current_device())
          - self._memstats.max_overall_cuda
      )
  
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 3 for calling the ChunkMemStatsCollector obj's underlying member method code with example usage: obj.cuda_margin_mem() and returns: unknown