模型结构可视化

通过 快速开始 一节，我们了解到，预测模型包含了两个文件，一部分为模型结构文件，通常以 pdmodel 文件存在；另一部分为参数文件，通常以 pdiparams 文件存在。

模型结构文件（*.pdmodel文件），顾名思义，存储了模型的拓扑结构，其中包括模型中各种OP的计算顺序以及OP的详细信息。很多时候，我们希望能够将这些模型的结构以及内部信息可视化，方便我们进行模型分析。

飞桨提供了 VisualDL 帮助我们来可视化模型结构，除此之外，对 Netron 熟悉的用户也可直接使用 Netron 进行模型结构的可视化。

安装可视化工具VisualDL

使用pip安装

python -m pip install visualdl -i https://mirror.baidu.com/pypi/simple

使用代码安装

git clone https://github.com/PaddlePaddle/VisualDL.git
cd VisualDL

python setup.py bdist_wheel
pip install --upgrade dist/visualdl-*.whl

可视化

支持两种启动方式：

• 前端拖拽上传模型文件：

  • 无需添加任何参数，在命令行执行 visualdl 后启动界面上传文件即可：

• 后端透传模型文件：

  • 在命令行加入参数 –model 并指定 模型文件 路径（非文件夹路径），即可启动：

visualdl --model model.pdmodel --port 8080

更多具体细节可参考 VisualDL使用指南