Keras:多个输出,损失仅是一个的函数吗?

时间:2019-07-15 23:43:23

标签: tensorflow keras

我有这样的设置:

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

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

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

我可以忍受这一点,但是我必须在各处查看此损失函数的统计信息和进度,并想知道是否还有更优雅的方法。

1 个答案:

答案 0 :(得分:2)

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

但是使用零损失也可以。