日々精進

新しく学んだことを書き留めていきます

Deep Learningでは基本乱数は完全には固定できない(なので全く同じモデルは作ることが難しい)

乱数のseedを固定すれば同じInputに対して、同じ推論結果を返す重みファイルが生成できるはず・・と思っていたが、そうではないらしい。

乱数の固定の仕方は以下参照。

qiita.com

これを設定しても、cuDNNの中の乱数は固定できないらしい。TensorFlowではこの問題の解決策は探した範囲では見付からず。。どうしてもやるならCPUで学習すればよい。でもメッチャ時間かかるからイヤダ・・

stackoverflow.com

qiita.com

PyTorchでは以下の方法でcuDNNの中も固定できるらしい。TensorFlowユーザなので試してないが・・

qiita.com