code: def _ignore_torch_cuda_oom():
      """
      A context which ignores CUDA OOM exception from pytorch.
      """
      try:
          yield
      except RuntimeError as e:
          # NOTE: the string may change?
          if "CUDA out of memory. " in str(e):
              pass
          else:
              raise
  
imports: import logging
  from contextlib import contextmanager
  from functools import wraps
  import torch
label_desc: Use image node_id 1 for calling a global function with example usage: _ignore_torch_cuda_oom() without return types