Fonction torch.use_deterministic_algorithms de PyTorch

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

torch.use_deterministic_algorithmsC'est une fonction de PyTorch pour utiliser des algorithmes déterministes. Elle définit s'il faut utiliser des algorithmes déterministes lorsque c'est possible.

Définition de fonction

torch.use_deterministic_algorithms(enabled, warn_only=False)

Exemple d'utilisation

Exemple

import torch

# Activer les algorithmes déterministes
torch.use_deterministic_algorithms(True)

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

# Désactiver les algorithmes déterministes
torch.use_deterministic_algorithms(False)
print(f"Utilisation d'algorithmes déterministes : {torch.are_deterministic_algorithms_enabled()}")

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

Autres extensions