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, выводить ошибки о недетерминированных операциях.