学习笔记|Pytorch使用教程19
本学习笔记主要摘自“深度之眼”,做一个总结,方便查阅。
使用Pytorch版本为1.2
- TensorBoard简介
- TensorBoard安装
- TensorBoard运行可视化
一.TensorBoard简介
TensorBoard : TensorFlow中强大的可视化工具
支持标量、图像、文本、音频、视频和Eembedding等多种数据可视化
二.TensorBoard安装
使用pip install tensorboard
安装注意事项
pip install tensorboard的时候会报错:
ModuleNotFoundError: No module named 'past'
通过pip install future
解决
三.TensorBoard运行可视化
测试代码
import numpy as np
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter(comment='test_tensorboard')
for x in range(100):
writer.add_scalar('y=2x', x * 2, x)
writer.add_scalar('y=pow(2, x)', 2 ** x, x)
writer.add_scalars('data/scalar_group', {"xsinx": x * np.sin(x),
"xcosx": x * np.cos(x),
"arctanx": np.arctan(x)}, x)
writer.close()
输出:
在当前路径下的终端中输入:tensorboard --logdir=./runs
在浏览器中打开网址:http://localhost:6006/
得到: