torch_geometric

class debug[source]

Context-manager that enables the debug mode to help track down errors and separate usage errors from real bugs.

Example

>>> with torch_geometric.debug():
...     out = model(data.x, data.edge_index)
is_debug_enabled()[source]

Returns True, if the debug mode is enabled.

class set_debug(mode)[source]

Context-manager that sets the debug mode on or off.

set_debug will enable or disable the debug mode based on its argument mode. It can be used as a context-manager or as a function.

See debug above for more details.