2

私の .theanorc ファイルで、パラメーターを設定しました...

[global]
floatX = float32

ただし、theano バックエンドで keras を実行し、model.predict を呼び出すと、返される配列の numpy データ型は常に FP32 ではなく FP64 型になります。これが問題なのか、それとも keras / theano が GPU で実行する前に FP32 に変換するのかはわかりません。確認する方法はありますか。GPU上でFP64を使用しようとすると、theanoが投稿してエラーまたは警告を出すことができれば幸いです。

4

1 に答える 1

3

タイプを確認するには、floatX単純に実行できます

import theano
print theano.config.floatX

そのコードが出力された場合、gpu 計算の入力として'float32'使用しようとすると、theano は警告を出力します。float64これは、キーワード引数を追加すると抑制できるため、グラフをコンパイルするときにallow_downcastこのキーワードが含まれていないことを確認してください。theano.function

于 2016-06-04T15:32:49.453 に答える