我正在使用自定义功能训练 Keras 模型,我之前已经成功测试过。最近,我用一个新的数据集训练它,我得到了一个奇怪的结果:模型训练得很好,但是val_loss
给出了nan
. 这是损失:
def Loss(y_true,y_pred):
y_pred = relu(y_pred)
z = k.maximum(y_true, y_pred)
y_pred_negativo = Lambda(lambda x: -x)(y_pred)
w = k.abs(add([y_true, y_pred_negativo]))
if k.sum(z) == 0:
error = 0
elif k.sum(y_true) == 0 and k.sum(z) != 0:
error = 100
elif k.sum(y_true) == 0 and k.sum(z) == 0:
error = 0
else:
error = (k.sum(w)/k.sum(z))*100
return error
我尝试了很多事情:
- 查看了 NaN 的数据
- 标准化 - 打开和关闭
- 剪辑 - 打开和关闭
- 辍学 - 打开和关闭
有人告诉我这可能是 CUDA 安装的问题,但我不确定。
关于问题是什么或我如何诊断它的任何想法?