1介绍
- tensorboard是tensorflow的可视化工具包,不过也可以在PyTorch中使用。
- [tensorflow]https://tensorflowcn.cn/tensorboard/get_started#google_vignette
- [torch.utils.tensorboard]https://pytorch.ac.cn/docs/2.6/tensorboard.html
2使用
- 安装与启动
pip install tensorboard
tensorboard --logdir=path/to/your/logdir
命令行启动tensorboard,参数指定的是tensorboard应该读取的日志文件夹路径,从这个文件中读取训练过程中生成的日志文件,从而实现可视化。
运行上述命令后,TensorBoard会在本地启动一个Web服务,通常默认监听6006端口。你可以通过浏览器访问以下地址来查看TensorBoard的可视化界面:
http://localhost:6006
- SummaryWriter提供tensorboard的可视化入口。
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter()
3源码大纲
class torch.utils.tensorboard.writer.SummaryWriter()
__init__()
add_scalar() # 添加标量数据
add_scalars()
add_histogram() # 直方图
add_image() # 添加图像数据,需要用pillow包(from PIL import Image)
add_images()
add_figure() # matplotlib
add_video() # moviepy
add_audio()
add_text()
add_graph()
add_embedding()
add_pr_curve() # 添加精确率-召回率曲线
add_custom_scalars()
add_mesh()
add_hparams()
flush()
close()
4使用注意事项
https://blog.youkuaiyun.com/u012010729/article/details/104330444
在使用命令tensorboard --logdir=logs时,要cd到logs文件夹的上层文件夹。
点开链接后,变成下方图示。有编码报错,应该是路径中有中文或者非下划线的字符。但是我不是这种情况,还是显示如下。
https://blog.youkuaiyun.com/bulletstart/article/details/134931659