Pytorch笔记--TensorBoard的使用

tensorboard简单介绍

TensorBoard是Google开发的一个机器学习可视化工具。其主要用于记录机器学习过程

 tensorboard的简单使用

一.绘制数学图像

from torch.utils.tensorboard import SummaryWriter


writer = SummaryWriter("logs") #logs是生成的图像所在文件夹名

print(type(img_array))
print(img_array.shape)

for i in range(100):
    writer.add_scalar("y=x", i, i)
writer.close()

这里绘制了一个 y=x的数学图像，用到的是SummaryWriter类中的add_scalar函数。

根据函数的文档字符串中可以了解到

“tag” -- 标题，在这里是我绘制的这个一元一次方程的表达式

“global_step” -- x轴

“global_step” -- y轴

 代码运行之后右侧项目文件夹就会多一个“logs”文件夹了，要打开tensorboard的窗口查看我们的图像需要在终端里敲指令