前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >TensorFlow保存TensorBoard图像操作

TensorFlow保存TensorBoard图像操作

作者头像
砸漏
发布2020-10-21 15:10:39
1.2K0
发布2020-10-21 15:10:39
举报
文章被收录于专栏:恩蓝脚本

简单的代码:

代码语言:javascript
复制
import tensorflow as tf

In [2]: matrix1=tf.constant([[3.,3.]])
In [3]: matrix2=tf.constant([[2.],[2.]])

with tf.Session() as sess:
  ...:   writer = tf.summary.FileWriter('./graph', sess.graph)
  ...:   result = sess.run(tf.matmul(matrix1, matrix2))
  ...:   writer.close()

ipython中使用!+命令可以直接运行terminal命令。

terminal输入: tensorboard –logdir graph/

跳出:Starting TensorBoard 54 at http://amax:6006

在浏览器输入地址加端口号并在graph中查看。

补充知识:tensorflow 利用保存的meta图文件生成log供tensorboard可视化 保存恢复模型

tensorboard可视化图:

代码语言:javascript
复制
import tensorflow as tf
g = tf.Graph() 
with g.as_default() as g: 
  tf.train.import_meta_graph('criteo_80.meta') 
 
with tf.Session(graph=g) as sess: 
  file_writer = tf.summary.FileWriter(logdir='./', graph=g)

保存恢复模型:

代码语言:javascript
复制
# 建模型
saver = tf.train.Saver()
 
with tf.Session() as sess:
  # 存模型,注意此处的model是文件名非路径
  saver.save(sess, "/tmp/model")
 
with tf.Session() as sess:
  # 恢复模型
  saver.restore(sess, "/tmp/model")
代码语言:javascript
复制
# 先恢复图
saver = tf.train.import_meta_graph("/tmp/model.meta")
 
with tf.Session() as sess:
  # 再恢复参数
  saver.restore(sess, "/tmp/model")

以上这篇TensorFlow保存TensorBoard图像操作就是小编分享给大家的全部内容了,希望能给大家一个参考。

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2020-09-11 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档