keras 如何添加tensorflow 损失函数?

wko9yo5t  于 2023-01-09  发布在  其他
关注(0)|答案(1)|浏览(142)

我不能把这两个损失相加如下

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

错误为:
TypeError:+的操作数类型不受支持:"二元互熵"与"二元互熵"

332nm8kg

332nm8kg1#

您可以将它们作为model.compile中的多个损失相加

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

这转换为final_loss = l1*loss1 + l2*loss2。只需将l1l2设置为1。

相关问题