Fonction torch.set_deterministic_debug_mode de PyTorch

Manuel de référence Pytorch torch

torch.set_deterministic_debug_modeCette fonction est utilisée pour définir le mode de débogage déterministe dans PyTorch. Lorsqu'elle est définie sur une valeur non nulle, PyTorch génère une erreur ou un avertissement lorsqu'un algorithme non déterministe est utilisé.

Définition de fonction

torch.set_deterministic_debug_mode(debug_mode)

Description du paramètre

  • debug_mode: Mode de débogage, valeurs facultatives :
    • 0: Désactivé (par défaut)
    • 1: Activer le mode strict
    • 2: Activer le mode d'avertissement

Exemple d'utilisation

Exemple

import torch

# Définir le mode de débogage déterministe sur le mode strict
# Cela lèvera une erreur lorsque des opérations non déterministes sont utilisées
torch.set_deterministic_debug_mode(1)

# Définir sur le mode d'avertissement
torch.set_deterministic_debug_mode(2)

# Désactiver le mode de débogage
torch.set_deterministic_debug_mode(0)

print("Le mode de débogage déterministe a été défini")

La sortie est :

Deterministic debug mode has been set

Manuel de référence Pytorch torch

Autres extensions