neural-network - 如何检查 theano 是否仅使用 FP32？

Question

在我的 .theanorc 文件中，我设置了参数...

[global]
floatX = float32

但是，当我使用 theano 后端运行 keras 并调用 model.predict 时，返回数组的 numpy 数据类型始终是 FP64 而不是 FP32 类型。我不确定这是否是一个问题，或者 keras / theano 是否在 GPU 上执行之前转换为 FP32。有没有办法检查。如果我尝试在 GPU 上使用 FP64，theano 可以发布错误或警告，我希望它。

Answer 1

要检查您的类型，floatX您可以简单地运行

import theano
print theano.config.floatX

如果该代码打印，'float32'那么当您尝试float64用作 gpu 计算的输入时，theano 将打印出警告。但是，如果您添加关键字参数，这可以被抑制allow_downcast，因此请确保在theano.function编译图形时没有此关键字。