Как проверить, что theano использует только FP32?

Question

В моем файле .theanorc я установить параметр ...

[global] 
floatX = float32 

Однако, когда я бег keras с Theano бэкэндом и делать звонки model.predict в Numpy типа данных возвращаемого массива всегда типа FP64 не FP32. Я не уверен, что это проблема, или если keras/anano делает преобразование в FP32 перед выполнением на GPU. Есть ли способ проверить. Мне бы хотелось, если бы anano мог опубликовать сообщение и ошибку или предупреждение, если я попытаюсь использовать FP64 на графическом процессоре.

Answer 1

Чтобы проверить тип floatX вы можете просто запустить

import theano 
print theano.config.floatX 

Если этот код печатает 'float32' то Theano выведет предупреждение при попытке использовать float64 в качестве входных данных для вычислений на GPU. Это может быть подавлено, но если вы добавите аргумент ключевого слова allow_downcast, убедитесь, что вы не имеете это ключевое слово в theano.function при компиляции графика.