pytorchでtorch.manual_seed()の理解

401 ワード

Pytoch python

次の操作を行います.

CPUにシードを設定し、乱数を生成

torch.manual_seed( number )

特定GPUにシードを設定し、乱数を生成

torch.cuda.manual_seed( number )

すべてのGPUにシードを設定し、乱数を生成

torch.cuda.manual_seed_all()

使用理由:

ランダムデータを生成する必要がある実験では、実験のたびにデータを生成する必要がある.ランダムシードを設定するのは,一定の乱数が生成されるたびに,実験結果が一致することを示し,実験の比較と改善に有利であることを確保するためである.