torch.set_deterministic_debug_mode

torch.set_deterministic_debug_mode(debug_mode) [source]

Устанавливает режим отладки для детерминированных операций.

Примечание

Это альтернативный интерфейс для torch.use_deterministic_algorithms(). Подробную информацию об охватываемых операциях см. в документации этой функции.

Параметры

debug_mode (str или int) – Если “default” или 0, не выводить ошибки или предупреждения при не детерминированных операциях. Если “warn” или 1, выводить предупреждения при не детерминированных операциях. Если “error” или 2, выводить ошибки при не детерминированных операциях.