gnn_tracking.utils.oom#

Attributes#

logger

N_OOM_ERRORS

Functions#

is_oom_error(→ bool)

Is this an out of memory (OOM) error?

tolerate_some_oom_errors(fct)

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

Module Contents#

gnn_tracking.utils.oom.logger#
gnn_tracking.utils.oom.N_OOM_ERRORS#
gnn_tracking.utils.oom.is_oom_error(e: Exception) bool#

Is this an out of memory (OOM) error?

gnn_tracking.utils.oom.tolerate_some_oom_errors(fct: Callable)#

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