MegEngine 使用小技巧:使用 Netron 实现模型可视化

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

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

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

如下图:

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

详细操作指南见文档:

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

但在实操时,有几点需要注意的地方: TracedModule 格式 Netron 需要是 6.0.0 及以上版本,为了保证最优体验效果,建议使用最新版本;

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

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

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

MegEngine 版本:v1.10.0 及以上;

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

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

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

关于 MyBatis-Flex 抄袭 MyBatis-Plus 的澄清 Arc 浏览器正式发布 1.0,声称是 Chrome 的替代品 OpenAI 正式上线安卓版 ChatGPT VS Code 优化名称混淆压缩,将内置 JS 减小 20%! LK-99:第一种室温常压超导体? 马斯克“零元购”,强夺 @x 推特账号 Python 指导委员会计划接受 PEP 703 提案,让全局解释器锁成为可选 特斯拉中国商城上架 1TB 固态硬盘,售价 2720 元 ProxyPin - 全平台系统开源免费抓包软件 Stack Overflow 访问量大幅下降,马斯克称其已被 LLM 取代
{{o.name}}
{{m.name}}

猜你喜欢

转载自my.oschina.net/u/5265910/blog/8184353
今日推荐