MegEngine 模型可视化¶
借助 Netron平台 ,我们可以查看 MegEngine 训练出的模型网络结构。Netron 是一种用于神经网络、深度学习和机器学习模型的可视化工具,它可以在 Linux、Windows 和 Mac 上跨平台运行,并且支持多种框架和格式。在2022年8月,我们将 MegEngine 集成在了此平台上,目前 Netron 已支持 MegEngine 模型的 TracedModule 及 C++ 计算图两种结构:
TracedModule 格式:
Netron 版本:6.0.0 及以上版本,为了保证最优体验效果,Netron 建议使用最新版本
保存 TracedModule 模型文件时需要用 .tm 作为文件后缀,推荐使用 megengine.save 和 megengine.load 保存和加载 TracedModule
因为 tm 格式的模型有很多子 module ,为了让大家更好的了解他们之间的链接关系,所以在做可视化展示时,各子图是全部展开的,以一张图来完整展示模型结构
C++ 格式:
Netron 版本:6.5.3 及以上版本,为了保证最优体验效果,Netron 建议使用最新版本
MegEngine 版本:v1.10.0 及以上
使用 Netron 进行 MegEngine 模型可视化¶
.tm 模型获取可参考 快速上手Traced Module , .mge 模型获取可参考 获得用于 MegEngine Lite 推理的模型 。 以.tm 模型为例,获取到模型后有三种方式可视化此模型。
在线网站¶
netron 作者很贴心地做了一个 在线demo网站 ,可以直接上传模型文件查看可视化结果,能够得到完整的模型结构图,并点击查看每层结构的细节参数。
Python Server¶
将 netron 作为 python 库进行安装,在 python 代码调用 netron 库来载入模型进行可视化。可以通过``pip install netron`` 进行安装。 在命令行输入:
netron Filename
点击回车,会生成地址,浏览器打开该地址,就出现我们训练的模型可视图了。
或者新建一个modelView.py文件,代码如下:
import netron # 使用netron进行模型可视化
modelPath = "./traced_shufflenet.tm" # 模型的路径
netron.start(modelPath) # 启动模型
随后运行modelView.py文件,生成地址,浏览器打开该地址,就出现我们训练的模型可视图了。
使用 Netron 客户端¶
以软件的方式安装netron,地址见 Netron官方github ,下载对应mac或windows对应版本的nerton软件,打开软件运行即可;
可视化效果¶
.tm 模型:
.mge 模型:
这里提供给大家 示例模型 供大家练习。