Функция PyTorch torch.set_deterministic_algorithm_enabled

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.set_deterministic_algorithm_enabledЭто функция в PyTorch, используемая для настройки детерминированных алгоритмов. Она определяет, включать ли детерминированные алгоритмы; при включении каждый запуск будет давать одинаковый результат.

Определение функции

torch.set_deterministic_algorithm_enabled(enabled, mode='warn')

Пример использования

Пример

import torch

# Установите, чтобы включить детерминированные алгоритмы
torch.set_deterministic_algorithm_enabled(True)

# Проверить, включено ли это
print(f"Детерминированность: {torch.are_deterministic_algorithms_enabled()}")

# Отключить детерминированные алгоритмы
torch.set_deterministic_algorithm_enabled(False)
print(f"Детерминированность: {torch.are_deterministic_algorithms_enabled()}")

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения