MegEngine 使用小技巧:使用 Netron 實現(xiàn)模型可視化
近期社區(qū)有多個同學問,如何查看 MegEngine 訓練出的模型網(wǎng)絡結構。其實在去年 8 月,MegEngine 就已經(jīng)集成到了 Netron 平臺上:https://netron.app/。
目前 Netron 已支持 MegEngine 模型:TracedModule 及 C++ 計算圖兩種結構。
二者的實現(xiàn)方法完全一致:只需直接拖拽本地模型文件到?網(wǎng)頁版 netron,便可得到完整的模型結構圖,并點擊查看每層結構的細節(jié)參數(shù)。
如下圖:

當然大家也可以通過「netron 客戶端」或「Python Server」的方式,得到同樣結果。
詳細操作指南見文檔:
https://www.megengine.org.cn/doc/stable/zh/user-guide/tools/viewmodel.html
但在實操時,有幾點需要注意的地方:
TracedModule 格式
Netron 需要是 6.0.0 及以上版本,為了保證最優(yōu)體驗效果,建議使用最新版本;
保存 TracedModule 模型文件時需要用 .tm 作為文件后綴,推薦使用?megengine.save?(https://www.megengine.org.cn/doc/stable/zh/reference/api/megengine.save.html#megengine.save)和?megengine.load(https://www.megengine.org.cn/doc/stable/zh/reference/api/megengine.load.html#megengine.load)?保存和加載 TracedModule;
因為 tm 格式的模型有很多子 module ,為了讓大家更好的了解他們之間的鏈接關系,所以在做可視化展示時,各子圖是全部展開的,以一張圖來完整展示模型結構。
C++ 格式
Netron 使用 6.5.3 及以上版本,為了保證最優(yōu)體驗效果,建議使用最新版本;
MegEngine 版本:v1.10.0 及以上;
歡迎大家通過以下樣例模型,試用該功能~
TracedModule 序列化文件(即 .tm 文件):?樣例模型()
C++ 序列化文件(即 .mge):樣例模型(https://www.megengine.org.cn/doc/stable/zh/reference/api/megengine.load.html#megengine.load)
附:
更多 MegEngine 信息獲取,您可以:查看文檔:https://www.megengine.org.cn/doc/stable/zh/
GitHub 項目: https://github.com/MegEngine
加入 MegEngine 用戶交流 QQ 群:1029741705
歡迎參與 MegEngine 社區(qū)貢獻,成為?Awesome MegEngineer:https://www.megengine.org.cn/community-AMGE,榮譽證書、定制禮品享不停