近期社区有多个同学问,如何查看 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 格式
-
Netron 需要是 6.0.0 及以上版本,为了保证最优体验效果,建议使用最新版本;
-
保存 TracedModule 模型文件时需要用 .tm 作为文件后缀,推荐使用