PyTorch torch.set_deterministic_algorithm_enabled Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.set_deterministic_algorithm_enabledEs ist eine Funktion in PyTorch, mit der deterministische Algorithmen festgelegt werden. Sie legt fest, ob deterministische Algorithmen aktiviert werden; wenn aktiviert, liefert jeder Lauf dasselbe Ergebnis.

Funktionsdefinition

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

Anwendungsbeispiel

Beispiel

import torch

# Deterministische Algorithmen aktivieren
torch.set_deterministic_algorithm_enabled(True)

# Prüfen, ob es aktiviert ist
print(f"Deterministisch: {torch.are_deterministic_algorithms_enabled()}")

# Deterministische Algorithmen deaktivieren
torch.set_deterministic_algorithm_enabled(False)
print(f"Deterministisch: {torch.are_deterministic_algorithms_enabled()}")

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Weitere Erweiterungen