Función PyTorch torch.set_deterministic_algorithm_enabled

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.set_deterministic_algorithm_enabledEs una función en PyTorch utilizada para configurar algoritmos deterministas. Establece si se habilita el algoritmo determinista; una vez habilitado, cada ejecución producirá el mismo resultado.

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

# Habilitar el algoritmo determinista
torch.set_deterministic_algorithm_enabled(True)

# Comprobar si está habilitado
print(f"Is deterministic: {torch.are_deterministic_algorithms_enabled()}")

# Deshabilitar el algoritmo determinista
torch.set_deterministic_algorithm_enabled(False)
print(f"Is deterministic: {torch.are_deterministic_algorithms_enabled()}")

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones