如何在Pytorch中使用TensorBoard

前言

TensorBoard是TensorFlow中强大的可视化工具

TensorBoard的工作流程：
1.python脚本记录可视化的数据
2.将数据存储到硬盘保存为event file
3.终端在网站输出结果

安装TensorBoard

激活环境输入

pip install tensorboard

接着输入

pip install future

都安装完成后可以输入

pip list

来查看是否成功安装这2个包。

代码演示

运行下方代码

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=./

在当前文件夹下即会产生一个event file，点击链接即可进入可视化的网站界面。