0

Theano 機械学習プログラムを作成しました。しかし、CPU と GPU ではまったく異なる 2 つの結果が得られました。

以下はログです。 (ログのごく一部のみ)

GPU での結果

CPU での結果

損失関数は急速に減少し、CPU で 0.2 に収束します。

ただし、損失関数は増加し、GPU では最終的に NaN になります。

私のプログラムにはどのような間違いがありますか? または、何に注意すればよいですか?ありがとうございました!

4

2 に答える 2

0

CPU がfloat64(倍精度) を使用し、GPU がfloat32(単精度) を使用している可能性はありますか? ここで設定フラグを調べることができます: http://deeplearning.net/software/theano/library/config.html

于 2016-03-29T10:48:57.037 に答える