使用Tensorflow后端在Keras训练时打印初始丢失

时间:2017-09-20 00:25:54

标签: tensorflow neural-network deep-learning keras

我在Keras(TF后端)训练我的Deep神经网络。我只是想在训练DNN时打印第一个损失。我只是想确保我的初始化是正确的,因此我需要在进行第一次正向传递后由DNN计算的初始损失。

Keras回调允许我们确定每个纪元后的损失。在第一次训练后我想要它。

1 个答案:

答案 0 :(得分:0)

您可以使用on_batch_end

创建自定义回调

https://keras.io/callbacks/