code: def _bytes_to_mb_gb(num_bytes: int) -> str:
      if num_bytes < 1024 * 1024:
          return f"{round(num_bytes / (1024 * 1024), 2)} MB"
      else:
          return f"{round(num_bytes / (1024 * 1024 * 1024), 2)} GB"
  
file_path: tnt/torchtnt/utils/oom.py
imports: import logging
  import os
  import pickle
  from typing import Any, Callable, Dict, Optional, Union
  import torch
  from torchtnt.utils.distributed import get_global_rank
  from torchtnt.utils.fsspec import get_filesystem
  from torchtnt.utils.version import is_torch_version_geq_2_0
label_desc: Use image node_id 4 for calling a global function with example usage: _bytes_to_mb_gb(num_bytes) and returns: round, str, round, str