gnn_tracking.utils.oom

Attributes

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.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.