torch.set_deterministic_debug_mode

torch.set_deterministic_debug_mode(debug_mode) [исходный код]

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

Примечание

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

Параметры:

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