code: def gpu_mem_usage():
      """Computes the GPU memory usage for the current device (MB)."""
      mem_usage_bytes = torch.cuda.max_memory_allocated()
      return mem_usage_bytes / 1024 / 1024
  
imports: from collections import deque
  import numpy
  import pycls.core.logging
  import torch
  from pycls.core.config import cfg
  from pycls.core.timer import Timer
label_desc: Use image node_id 3 for calling a global function with example usage: gpu_mem_usage() and returns: unknown