PaddleClas/docs/zh_CN/extension/VisualDL.md

1.7 KiB
Raw Blame History

使用VisualDL可视化训练过程

前言

VisualDL是飞桨可视化分析工具以丰富的图表呈现训练参数变化趋势、模型结构、数据样本、高维数据分布等。可帮助用户更清晰直观地理解深度学习模型训练过程及模型结构进而实现高效的模型优化。更多细节请查看VisualDL

在PaddleClas中使用VisualDL

现在PaddleClas支持在训练阶段使用VisualDL查看训练过程中学习率learning rate、损失值loss以及准确率accuracy的变化情况。

设置config文件并启动训练

在PaddleClas中使用VisualDL只需在训练配置文件config文件添加如下字段

# config.yaml
vdl_dir: "./vdl.log"

vdl_dir 用于指定VisualDL用于保存log信息的目录。

然后正常启动训练即可:

python3 tools/train.py -c config.yaml

启动VisualDL

在启动训练程序后可以在新的终端session中启动VisualDL服务

 visualdl --logdir ./vdl.log

上述命令中,参数--logdir用于指定日志目录VisualDL将遍历并且迭代寻找指定目录的子目录将所有实验结果进行可视化。也同样可以使用下述参数设定VisualDL服务的ip及端口号

  • --host设定IP默认为127.0.0.1
  • --port设定端口默认为8040

更多参数信息,请查看VisualDL

在启动VisualDL后即可在浏览器中查看训练过程输入地址127.0.0.1:8840