Fonction PyTorch torch.set_deterministic_algorithm_enabled

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.set_deterministic_algorithm_enabledIl s'agit d'une fonction de PyTorch utilisée pour définir des algorithmes déterministes. Elle définit s'il faut activer les algorithmes déterministes ; lorsqu'ils sont activés, chaque exécution produira le même résultat.

Définition de fonction

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

Exemple d'utilisation

Exemple

import torch

# Définir pour activer les algorithmes déterministes
torch.set_deterministic_algorithm_enabled(True)

# Vérifier s'il est activé
print(f"Est déterministe : {torch.are_deterministic_algorithms_enabled()}")

# Désactiver les algorithmes déterministes
torch.set_deterministic_algorithm_enabled(False)
print(f"Est déterministe : {torch.are_deterministic_algorithms_enabled()}")

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions