PyTorch torch.set_deterministic_debug_mode 関数

Pytorch torch リファレンスマニュアル

torch.set_deterministic_debug_modeは PyTorch において決定性デバッグモードを設定するための関数です。非ゼロ値に設定すると、PyTorch は非決定的アルゴリズムを使用した際にエラーまたは警告を発生させます。

関数定義

torch.set_deterministic_debug_mode(debug_mode)

パラメータ説明

  • debug_mode: デバッグモード。選択可能な値:
    • 0: オフ（デフォルト）
    • 1: 厳格モードを有効にする
    • 2: 警告モードを有効にする

使用例

例

import torch

# 決定性デバッグモードを厳格モードに設定
# これにより非決定的操作を使用した際にエラーが発生します
torch.set_deterministic_debug_mode(1)

# 警告モードに設定
torch.set_deterministic_debug_mode(2)

# デバッグモードをオフにする
torch.set_deterministic_debug_mode(0)

print(「決定性デバッグモードが設定されました」)

出力結果は次のとおりです:

決定性デバッグモードが設定されました

Pytorch torch リファレンスマニュアル

その他の拡張