学习笔记|Pytorch使用教程19(TensorBoard简介与安装)

学习笔记|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/
得到：