思考の本棚

機械学習のことや読んだ本の感想を整理するところ

2021-01-04から1日間の記事一覧

PyTorch-Lightningでの再現性

はじめに PyTorchで再現性を持たせたい場合、lightningのseed_everything関数や以下のような自作関数を使うケースが多いと思います。 def set_seed(seed: int = 42): random.seed(seed) np.random.seed(seed) os.environ["PYTHONHASHSEED"] = str(seed) torc…