如何在keras的自定义损失函数中调用全局变量以更改损失函数的返回值

时间:2019-04-12 13:40:14

标签: python-3.x tensorflow keras

我想定义一个损失函数,而y_true的损失是我不打算使用的,但我将从外部全局变量中得到的结果,我发现在传递定义的损失函数后到模型编译时,它只被调用了一次,我以为他要调用一次batch_size,因为我在自定义损失函数中添加了打印,但只打印了一次,所以我想知道自定义损失函数有多少次被称为。

def my_loss(y_true, y_pred):

    global loss_count
    loss_count += 1
    print("the loss",loss_count)
    return keras.losses.categorical_crossentropy(y_true, y_pred)

模型训练后,可变损失计数等于1

0 个答案:

没有答案