博客

MegEngine 使用小技巧:使用 Netron 实现模型可视化
作者:MegEngine 发布日期:2023/02/23

近期社区有多个同学问,如何查看 MegEngine 训练出的模型网络结构。其实在去年 8 月,MegEngine 就已经集成到了 Netron 平台上。

目前 Netron 已支持 MegEngine 模型:TracedModule 及 C++ 计算图两种结构。

二者的实现方法完全一致:只需直接拖拽本地模型文件到 网页版 netron,便可得到完整的模型结构图,并点击查看每层结构的细节参数。

如下图:

image.png

当然大家也可以通过「netron 客户端」或「Python Server」的方式,得到同样结果。

详细操作指南见文档:

MegEngine 模型可视化 - MegEngine 1.12 文档

但在实操时,有几点需要注意的地方:

TracedModule 格式

  1. Netron 需要是 6.0.0 及以上版本,为了保证最优体验效果,建议使用最新版本;

  2. 保存 TracedModule 模型文件时需要用 .tm 作为文件后缀,推荐使用 megengine.savemegengine.load 保存和加载 TracedModule;

  3. 因为 tm 格式的模型有很多子 module ,为了让大家更好的了解他们之间的链接关系,所以在做可视化展示时,各子图是全部展开的,以一张图来完整展示模型结构。

C++ 格式

  1. Netron 使用 6.5.3 及以上版本,为了保证最优体验效果,建议使用最新版本;

  2. MegEngine 版本:v1.10.0 及以上;

欢迎大家通过以下样例模型,试用该功能~

附:

更多 MegEngine 信息获取,您可以:查看文档GitHub 项目,或加入 MegEngine 用户交流 QQ 群:1029741705。欢迎参与 MegEngine 社区贡献,成为 Awesome MegEngineer,荣誉证书、定制礼品享不停。

查看更多

上一篇: 分享

MegCC 新版本来啦!新增 Benchmark 等工具,性能大幅提升!有奖征文活动同步启动!

下一篇: 分享

mperf:移动/嵌入式平台算子性能调优利器

相关推荐
  • MegEngine Python 层模块串讲(中)

    2023/07/28

  • BaseDet: 走过开发的弯路

    2022/12/15

  • 金枪鱼之夜:MegEngine 框架设计

    2020/08/23

  • 旷视天元Beta版核心技术升级全面解读

    2020/07/24

  • MegEngine Windows Python wheel 包减肥之路

    2022/01/05