Функция PyTorch torch.set_deterministic_debug_mode
Справочное руководство Pytorch torch
torch.set_deterministic_debug_modeЭта функция используется для установки режима отладки детерминированности в PyTorch. Если задано ненулевое значение, PyTorch выдаст ошибку или предупреждение при использовании недетерминированного алгоритма.
Определение функции
torch.set_deterministic_debug_mode(debug_mode)
Описание параметра
debug_mode: Режим отладки, возможные значения:0: Выключено (по умолчанию)1: Включить строгий режим2: Включить режим предупреждений
Пример использования
Пример
import torch
# Установить режим отладки детерминированности в строгий режим
# Это вызовет ошибку при использовании недетерминированных операций
torch.set_deterministic_debug_mode(1)
# Установить режим предупреждений
torch.set_deterministic_debug_mode(2)
# Отключить режим отладки
torch.set_deterministic_debug_mode(0)
print("Режим отладки детерминированности установлен")
# Установить режим отладки детерминированности в строгий режим
# Это вызовет ошибку при использовании недетерминированных операций
torch.set_deterministic_debug_mode(1)
# Установить режим предупреждений
torch.set_deterministic_debug_mode(2)
# Отключить режим отладки
torch.set_deterministic_debug_mode(0)
print("Режим отладки детерминированности установлен")
Вывод:
Deterministic debug mode has been set
Другие расширения