基于VisualDL的可视化方法分析
文章目录
前言
本文主要面向刚刚接触paddlepaddle框架,希望使用可视化方法调参或者查看loss情况的读者。利用飞桨自带的VisualDL可以方便的在每个mini batch 之后在图像中画点,查看趋势
一、VisualDL安装
在cmd命令行中输入:
pip install --upgrade --pre visualdl
出现Successfully installed visualdl-2.0.0证明安装成功
二、使用步骤
1.引入库
引入VisualDL库,定义作图数据存储位置(供第3步使用),本案例的路径是“log”:
from visualdl import LogWriter
log_writer = LogWriter("./log")
2.设置mini batch,画点
在训练过程中插入作图语句。当每100个batch训练完成后,将当前损失作为一个新增的数据点(iter和acc的映射对)存储到第一步设置的文件中。使用变量iter记录下已经训练的批次数,作为作图的X轴坐标。
log_writer.add_scalar(tag = 'acc', step = iter, value = avg_acc.numpy())
log_writer.add_scalar(tag = 'loss', step = iter, va