本文基于上面这篇文章,将打印出的损失函数值和预测值进行图像化的显示
实现的步骤
- 在张量中定义一个汇总的张量
self.precise_summary = tf.summary.scalar('precise',self.precise)self.loss_summary = tf.summary.scalar("loss",self.loss) - 在训练中定义一个filewriter(指明计算图)
writer = tf.summary.FileWriter(cfg.logdir, self.session.graph) - 在训练的时候计算汇总,查看汇总的值,加入到filewriter中
writer.add_summary(loss_summary, epoch*batches+batch)writer.add_summary(precise_summary, epoch*batches) - 新建并定义logdir目录
self.logdir = '../logs/P35/' - 启动tensorboard
- 打开浏览器查看