乐趣区

关于深度学习:MegEngine-使用小技巧使用-Netron-实现模型可视化

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

目前 Netron 已反对 MegEngine 模型:TracedModule 及 C++ 计算图两种构造。

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

如下图:

当然大家也能够通过「netron 客户端」或「Python Server」的形式,失去同样后果。

具体操作指南见文档:

MegEngine 模型可视化 – MegEngine 1.12 文档 www.megengine.org.cn/doc/stable/zh/user-guide/tools/viewmodel.html

但在实操时,有几点须要留神的中央:

TracedModule 格局

  1. Netron 须要是 6.0.0 及以上版本,为了保障最优体验成果,倡议应用最新版本;
  2. 保留 TracedModule 模型文件时须要用 .tm 作为文件后缀,举荐应用 megengine.save 和 megengine.load 保留和加载 TracedModule;
  3. 因为 tm 格局的模型有很多子 module,为了让大家更好的理解他们之间的链接关系,所以在做可视化展现时,各子图是全副开展的,以一张图来残缺展现模型构造。

C++ 格局

  1. Netron 应用 6.5.3 及以上版本,为了保障最优体验成果,倡议应用最新版本;
  2. MegEngine 版本:v1.10.0 及以上;

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

  • TracedModule 序列化文件(即 .tm 文件):样例模型
  • C++ 序列化文件(即 .mge):样例模型

附:

更多 MegEngine 信息获取,您能够:查看文档和 GitHub 我的项目,或退出 MegEngine 用户交换 QQ 群:1029741705。欢送参加 MegEngine 社区奉献,成为 Awesome MegEngineer,荣誉证书、定制礼品享不停。

退出移动版