PyTorch torch.set_deterministic_algorithm_enabled 関数

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

torch.set_deterministic_algorithm_enabledPyTorch において決定論的アルゴリズムを設定するための関数です。決定論的アルゴリズムを有効にするかどうかを設定し、有効にすると毎回同じ結果が得られます。

関数定義

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

使用例

実例

import torch

# 決定論的アルゴリズムを有効にする
torch.set_deterministic_algorithm_enabled(True)

# 有効かどうかを確認する
print(f"Is deterministic: {torch.are_deterministic_algorithms_enabled()}")

# 決定論的アルゴリズムを無効にする
torch.set_deterministic_algorithm_enabled(False)
print(f"Is deterministic: {torch.are_deterministic_algorithms_enabled()}")

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

その他の拡張