torch_geometric

class debug

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()

Returns True, if the debug mode is enabled.

class set_debug(mode)

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.