在每个step打印的是训练集的metric,在验证集中打印的是loss
来源:3-12 tf.GradientTape与tf.keras结合使用
战战的坚果
2020-04-29
老师,这个小节中,实现自定义梯度后,在每个step打印的是训练集的metric,在验证集中打印的是loss。原来使用fit函数时,它每一次输出的是训练集的loss和验证集的是loss,它是怎么实现的呀。
写回答
1回答
-
其实思想是类似的,也是batch遍历,最后结果混总。
你可以看其中的training_arrays.ArrayLikeTrainingLoop()函数,然后再往底层走,就是model_iteration函数:https://github.com/tensorflow/tensorflow/blob/e5bf8de410005de06a7ff5393fafdf832ef1d4ad/tensorflow/python/keras/engine/training_arrays.py#L46
不过它实现的比较复杂,思想就是我上面说的。
012020-04-30
相似问题