PyTorch torch.set_deterministic_debug_mode Función

Manual de referencia de Pytorch torch

torch.set_deterministic_debug_modeEs una función en PyTorch que se utiliza para configurar el modo de depuración determinista. Cuando se establece en un valor distinto de cero, PyTorch lanzará un error o una advertencia al utilizar algoritmos no deterministas.

Definición de la función

torch.set_deterministic_debug_mode(debug_mode)

Descripción de parámetros

  • debug_mode: Modo de depuración, valores opcionales:
    • 0: Desactivado (predeterminado)
    • 1: Activar modo estricto
    • 2: Activar modo de advertencia

Ejemplo de uso

Ejemplo

import torch

# Configurar el modo de depuración determinista como modo estricto
# Esto lanzará un error al utilizar operaciones no deterministas
torch.set_deterministic_debug_mode(1)

# Configurar como modo de advertencia
torch.set_deterministic_debug_mode(2)

# Desactivar el modo de depuración
torch.set_deterministic_debug_mode(0)

print("El modo de depuración determinista se ha configurado")

El resultado de salida es:

Se ha configurado el modo de depuración determinista

Manual de referencia de Pytorch torch

Otras extensiones