使用TensorBoard
Posted birdmmxx
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了使用TensorBoard相关的知识,希望对你有一定的参考价值。
偶然机会需要使用ckpt转换为pb,所以需要查找到输出节点。
指定ckpt文件,然后会在当前文件夹下生成__tb文件夹,里面还有一个文件。
import tensorflow as tf sess = tf.Session() tf.train.import_meta_graph("F:/ModelArts/cityscapesScripts/exp/train_on_train_set/train/model.ckpt-1000.meta") tf.summary.FileWriter("__tb", sess.graph)
切换到指定目录
最后会有一个链接
在浏览器中复制打开
以上是关于使用TensorBoard的主要内容,如果未能解决你的问题,请参考以下文章
深度学习pytorch使用tensorboard可视化实验数据
深度学习pytorch使用tensorboard可视化实验数据
错误处理笔记 导入 torch.utils.tensorboard时 找不到tensorboard
运行代码时出现ModuleNotFoundError: No module named ‘tensorboard‘解决方法