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 os
  import sys
  import time
  import functools
  import logging
  from contextlib import contextmanager
  from functools import wraps
  import numpy
  import torch.nn
  import torch
label_desc: Use image node_id 11 for calling a global function with example usage: _ignore_torch_cuda_oom() without return types