首页 > 解决方案 > 如何添加张量流损失函数?

问题描述

我不能将这两个损失加起来如下

real_loss = tf.losses.BinaryCrossentropy(tf.ones_like(train_images[0]),train_images[0])
fake_loss = tf.losses.BinaryCrossentropy(tf.zeros_like(train_images[0]),train_images[0])
fake_loss+real_loss

错误是:

类型错误:+ 不支持的操作数类型:“BinaryCrossentropy”和“BinaryCrossentropy”

标签: tensorflowkerasloss-function

解决方案


您可以将它们添加为多个损失model.compile

model.compile(loss = [loss1,loss2], loss_weights = [l1,l2], ...)

这转化为final_loss = l1*loss1 + l2*loss2. 只需将l1and设置l2为 1。


推荐阅读