训练数据可视化：visualdl的使用

• 1、visualdl
• 2、安装
• 3、写入数据
• 4、如何使用
• 5、更多

1、visualdl

visualdl是百度pp飞浆出品的产品，其效果图如图，实际上和tensorboard差不多，本文简单记录下visualdl的使用，当然tensorboard也是这么使用的，举一反三而已。

2、安装

pip install --upgrade visualdl

3、写入数据

visualdl可写入的数据包括标量、图像、音频等等，这里介绍下标量的记录。
我们在训练模型时，每iter都会有各种loss和精度，这种就属于标量

首先，实例化一个对象：

from visualdl import LogWriter

writer=LogWriter(logdir="./log")
# logdir是指定记录的log文件存放位置

现在记录训练时的loss：

iter=0
for labels,imgs in dataloader:
	iter=iter+1
	predicts=model(imgs)
	loss=func(labels,predicts)
	writer.add_scalar(tag="loss", step=iter, value=loss)

参数含义如下图：

4、如何使用

比如上方这个writer是定义为将log文件保存在log/文件夹中。
通过命令

visualdl logdir=log/

便可启动，注意：log/文件夹里是有log文件的。

在浏览器中访问：

http://localhost:8040/

即可

5、更多

更多用法可以移步visualDL的github：PaddlePaddle/VisualDL