tensorflow 1.2.0 中使用可视化tensorboard工具

来源:互联网 发布:阿里虚拟主机php mysql 编辑:程序博客网 时间:2024/05/22 06:12

执行完python文件后,启动tensorboard,会发现没有任何graph图展示,那就注意了

因为版本的不同,需要再py文件里增加语句:

writer = tf.train.SummaryWriter('./my_graph', sess.graph)-----一般这个语句低版本可以,像1.2.0就不行,报错:module doesn't have attribute 'SummaryWriter'

综合网上查找的:改写如下

merged = tf.summary.merge_all()
 writer = tf.summary.FileWriter('tmp', sess.graph) #自动创建文件夹tmp,与py文件在同一路径下
再次执行,tensorboard --logdir=tmp就可以在浏览器上看到想看到的graph了!


阅读全文
0 0
原创粉丝点击