torch.set_deterministic_debug_mode

torch.set_deterministic_debug_mode(debug_mode)

Sets the debug mode for deterministic operations.

Note

This is an alternative interface for torch.use_deterministic_algorithms(). Refer to that function’s documentation for details about affected operations.

Parameters

debug_mode (str or int) – If “default” or 0, don’t error or warn on nondeterministic operations. If “warn” or 1, warn on nondeterministic operations. If “error” or 2, error on nondeterministic operations.