PyTorch torch.set_deterministic_debug_mode Funktion

PyTorch torch Referenzhandbuch

torch.set_deterministic_debug_modeDiese Funktion wird verwendet, um den deterministischen Debug-Modus in PyTorch einzustellen. Wenn sie auf einen Wert ungleich null gesetzt wird, löst PyTorch einen Fehler oder eine Warnung aus, wenn ein nicht-deterministischer Algorithmus verwendet wird.

Funktionsdefinition

torch.set_deterministic_debug_mode(debug_mode)

Parameterbeschreibung

  • debug_mode: Debug-Modus, optionale Werte:
    • 0: Aus (Standard)
    • 1: Strikten Modus aktivieren
    • 2: Warnmodus aktivieren

Anwendungsbeispiel

Beispiel

import torch

# Den deterministischen Debug-Modus auf den strikten Modus setzen
# Dies löst einen Fehler aus, wenn nicht-deterministische Operationen verwendet werden
torch.set_deterministic_debug_mode(1)

# Auf Warnmodus setzen
torch.set_deterministic_debug_mode(2)

# Debug-Modus ausschalten
torch.set_deterministic_debug_mode(0)

print("Deterministischer Debug-Modus wurde gesetzt")

Die Ausgabe ist:

Deterministic debug mode has been set

PyTorch torch Referenzhandbuch

Andere Erweiterungen