matconvnet​/example/i​magenetにおけ​る、乱数の固定につい​て

1 view (last 30 days)
Hiroshi Kawaguchi
Hiroshi Kawaguchi on 30 Nov 2017
Answered: Naoya on 30 Nov 2017
matconvnet/example/imagenetにおいてGPUを用いての実行毎の結果に再現性を出したいのですが、可能でしょうか。 cnn_imagenet_init.m内のweightsの初期値はシード値を用いて、固定しています。 matconvnet内のm,cuファイル、を見たところ、どれも決定的な乱数を用いているように思うのですが、やはりcuDNN内で非決定的な値が使われているのでしょうか。 値を固定することが可能なら、ご教授いただけると幸いです。

Accepted Answer

Naoya
Naoya on 30 Nov 2017
MatConvNetで作成されるmファイルの詳細は把握しておりませんが、 Neural Network Toolbox 側でも trainNetwork() による GPU 上での計算では、 cuDNN を使っており、 cnDNN 側の非決定的なルーチンが原因で毎回計算結果が異なってきます。
下記は関連記事となります。

More Answers (0)

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!