gnn_tracking.utils.oom
======================

.. py:module:: gnn_tracking.utils.oom


Attributes
----------

.. autoapisummary::

   gnn_tracking.utils.oom.N_OOM_ERRORS


Functions
---------

.. autoapisummary::

   gnn_tracking.utils.oom.is_oom_error
   gnn_tracking.utils.oom.tolerate_some_oom_errors


Module Contents
---------------

.. py:data:: N_OOM_ERRORS

.. py:function:: is_oom_error(e: Exception) -> bool

   Is this an out of memory (OOM) error?


.. py:function:: tolerate_some_oom_errors(fct: Callable)

   Decorators to tolerate a couple of out of memory (OOM) errors.