class_docstr: Estimate memory with chunk
class_docstr_tok: ["Estimate","memory","with","chunk"]
docstr: sum all memory of active nodes
docstr_tok: ["sum","all","memory","of","active","nodes"]
code: def _get_memory_from_active_nodes(self, active_nodes: Dict) -> float:
      """
      sum all memory of active nodes
      """
      out = [i for i in active_nodes.values()]
      out = sum(out)
      return out
  
file_path: colossalai/colossalai/autochunk/estimate_memory.py
imports: from typing import Dict, List
  import torch
  from torch.fx.node import Node
  from .utils import NodeMgr, get_node_shape, is_non_memory_node
label_desc: Use image node_id 10 for calling the EstimateMemory obj's underlying member method code with example usage: obj._get_memory_from_active_nodes(active_nodes) and returns: out