Pytorchで再現性を保つ方法

概要

Pytorchでモデルを組みGPUで学習をさせると、学習させる度にLossが変わる。
このままだと再現性がない。

解決策

pytorch.org

公式によると、
学習させるコードに

torch.manual_seed(0)

を追記する。
もし、GPU（cuDNN）で学習させる場合には

cudnn.deterministic = True
cudnn.benchmark = False

の2行も追記する

参考

qrunch.net