mxnet模型可视化

使用netron

解析mxnet模型文件,可视化模型整体单元,但无法观察通道shape

pip install netron -i https://pypi.tuna.tsinghua.edu.cn/simple

netron直接运行,添加json文件即可

使用Graphviz,可视化

安装：

sudo apt-get install graphviz

可视化代码:

import mxnet as mx

sym,arg_params,aux_params =