python - Keras 中的自定义损失函数仅运行一次

Question

我在 Keras 中定义了一个自定义损失函数，如下所示:

def loss_function_test(yTrue, yPred):
    global i
    i += 1
    print("\n")
    print("Loss Function:", i, " ----- ", yTrue, yPred)
    print("\n")
    res = k.sum(yTrue - yPred)
    return res

并设置如下:

model.compile(loss=loss_function_test,
              optimizer='sgd',
              metrics=['accuracy'])

问题是检查损失函数代码中的global i和print函数，我意识到当我调用时，该函数在训练中仅被调用一次>模型.fit。即使我调试它，它也只被调用一次。看起来损失函数可能只在编译时被调用。谁能解释一下为什么吗？

Answer 1

构造损失函数 op 的函数仅运行一次。

如果您想查看 tensorflow 评估操作的频率，您需要添加 tf.Print运算到 tensorflow 图中。