是否可以通过在Keras中将model.compile中的loss_weights设置为零来删除损失函数项?

时间:2019-07-20 08:21:44

标签: keras keras-layer tf.keras

我的全损函数有三个术语:

L = λ1*L1 + λ2*L2 + λ3*L3

当我运行model.compile时,所有的λ由loss_weights{"λ1":1, "λ2":1, "λ2":1}设置。
现在,我要删除L1一词。
如果我在loss_weights中更改loss_weights{"λ1":0, "λ2":1, "λ2":1}而不是在模型中删除L1项的输出,可以吗?

1 个答案:

答案 0 :(得分:0)

是的,应该没问题,它将抵消损失的那部分的梯度。这个技巧通常是在对象检测损失中完成的,因此我们知道它是有效的。