Keras:损失函数在没有GPU的情况下会失效,而在GPU的情况下保持数值

时间:2018-11-02 17:44:00

标签: keras gpu overflow underflow numerical-stability

我对Keras中的上溢/下溢有一个基本问题。在没有GPU的主机上,具有相同负载权重的相同模型应该变为NaN,但在具有GPU的主机上继续训练就可以了吗?

在后者上,我的模型可以继续使用GPU进行训练,但是在前者上,我始终会出现不足/溢出,从而导致NaN。我想弄清楚如何在没有GPU的主机上继续训练,因为这对特定任务很有帮助。

非常欢迎其他人对这个问题的建议!

0 个答案:

没有答案
相关问题