code: def get_mem_info(prefix=""):
      return f"{prefix}GPU memory usage: {get_gpu_mem():.2f} MB, CPU memory usage: {get_cpu_mem():.2f} MB"
  
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 3 for calling a global function with example usage: get_mem_info(prefix) and returns: prefix, str, get_gpu_mem, str, get_cpu_mem, str