code: def get_cpu_mem():
      return psutil.Process().memory_info().rss / 1024**2
  
file_path: ColossalAI/examples/language/gpt/experiments/auto_parallel/auto_parallel_with_gpt.py
imports: from functools import partial
  from time import time
  import psutil
  import torch
  import transformers
  from gpt_modules import GPT2LMHeadModel, GPTLMLoss
  from colossalai.auto_parallel.tensor_shard.initialize import autoparallelize
  from colossalai.initialize import launch_from_torch
  from colossalai.legacy.core import global_context
  from colossalai.logging import disable_existing_loggers, get_dist_logger
label_desc: Use image node_id 1 for calling a global function with example usage: get_cpu_mem() and returns: unknown