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/community/roberta/pretraining/utils/exp_util.py
imports: import functools
  import os
  import shutil
  import psutil
  import torch
  from colossalai.legacy.core import global_context
label_desc: Use image node_id 6 for calling a global function with example usage: get_mem_info(prefix) and returns: prefix, str, get_gpu_mem, str, get_cpu_mem, str