TensorBoard记录loss变化

作者: admin 日期: 2017-10-11 13:50:58 人气: - 评论: 0

使用tensorboard来观察loss变化可以很好的进行调参

在定义好了loss之后调用:

tf.summary.scalar("loss",loss) #把loss加入summary中

merged = tf.summary.merge_all()#合并所有操作

train_writer = tf.summary.FileWriter("C:\\tmp\\tensorflow\mnist\log",

                                   sess.graph) #获得write对象


在训练的循环中调用,其中i是学习的步数

result = sess.run(merged, feed_dict= {x: x_train, y: y_train})
train_writer.add_summary(result,i)


结果如下:


相关内容

发表评论
更多 网友评论0 条评论)
暂无评论

Copyright © 2012-2014 我的代码板 Inc. 保留所有权利。

页面耗时0.0195秒, 内存占用1.82 MB, 访问数据库13次

闽ICP备15009223号-1