Visdom数据可视化工具绘制Loss曲线

Visdom介绍

Visdom是FaceBook公司开发的一款开源数据可视化工具，以其简单易用的功能，很快成为PyTorch的一个数据可视化工具。主要将其用来做loss的可视化还有一些图片的可视化功能。当然，使用Visdom工具不单单可以进行上述可视化展示，还可以进行更多的可视化展示。

通过pip命令安装好visdom之后，通过python -m visdom.server就可以打开visdom了。

Loss可视化代码编写

当然，最主要的还是进行loss可视化工作。可以进行实时的loss曲线绘制，之所以能够实现实时绘制曲线关键在于append。下面展示一下示例的代码：

from visdom import Visdom
import numpy as np
import time

x = np.linspace(1,50,500)
y = 1/x
y = y.reshape(500,1)
x = x.reshape(500,1)

vis = Visdom()

loss_window = vis.line(
    X=x[0],
    Y=y[0],
    opts={'xlabel': 'epochs', 'ylabel': 'loss_value', 'title': 'loss'}
)

for i in range(1, 500):
    time.sleep(0.1)
    vis.line(
        X=x[i],
        Y=y[i],
        win=loss_window,
        update='append'
    )

绘制效果如下：

Visdom还可以画很多其他类型的图表，具体讲解，还有代码如何使用，如果需要学习可以看我录制的一个课程视频

视频链接