首页 > 解决方案 > Keras:多个输出,只损失一个函数?

问题描述

我有这样的设置:

model = keras.Model(input,[output1,output2])

我的损失函数只是output1的函数。为了计算损失,我如何告诉 Keras 忽略output2 ?我想出的最好方法是生成一个始终返回 0.0 的虚假损失函数:

model.compile(optimizer=..., loss=[realLossFunction, zeroLossFunction])

我可以忍受这个,但是我必须到处查看这个损失函数的统计数据和进度,并且想知道是否有更优雅的方法。

标签: tensorflowkeras

解决方案


您可以避免将此输出放入模型中,然后重用权重(或与功能 API 共享)以将附加输出添加到完整模型中。

但是使用零损失也很好。


推荐阅读