用tensorboard实现scalars(图表)可视化。
文章目录
1.初始化日志文件
首先需要引入下面的代码,这些代码的位置在main.py里面。
其中,logdir是存放日志文件的目录。
import tensorflow as tf
logdir = './tensorboard/test1/'
summary_writer = tf.summary.create_file_writer(logdir)
项目结构如下:

2.构造图表
然后需要在训练的地方加上下述代码。
在这里,我是把loss给记录下来。step表示步长,是tensorboard折线图中的横坐标,一般用当前的epoch来表示。此处纵坐标就是loss值了。
with

文章介绍了如何使用Tensorboard对训练过程进行可视化,包括初始化日志文件、记录损失函数值并构造图表,以及解决在运行过程中遇到的无dashboard问题。关键步骤涉及设置logdir、创建file_writer、记录scalar数据,并通过指定端口号启动tensorboard服务。
最低0.47元/天 解锁文章
5337

被折叠的 条评论
为什么被折叠?



