tensorflow记录训练和验证的loss

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了tensorflow记录训练和验证的loss相关的知识,希望对你有一定的参考价值。

参考技术A 一般来说,我们会训练一个epoch然后验证一部分数据集。如何记录两部分的loss。
之前的写法主要是记录训练时的loss,所以是这样写的

相当于不停地监控model的loss,但一个Epoch之后,需要验证时,train_loss还在记录,这就不是很好了,查阅了很多资料,下面提供一种方案。
训练时

验证时

最后的效果:

以上是关于tensorflow记录训练和验证的loss的主要内容,如果未能解决你的问题,请参考以下文章

如何将 Tensorflow 数据集 API 与训练和验证集一起使用

训练和验证模式 tensorflow 的 SAME 数据丢失不一致

使用相同的图在 TensorFlow 中显示训练和验证的准确性

tensorflow在训练和验证时监视不同的summary的操作

如何拆分自己的数据集以在 Tensorflow CNN 中进行训练和验证

Tensorflow DataSet Shuffle 影响验证训练的准确性和模棱两可的行为