最美情侣中文字幕电影,在线麻豆精品传媒,在线网站高清黄,久久黄色视频

歡迎光臨散文網(wǎng) 會員登陸 & 注冊

MegEngine 使用小技巧:使用 Netron 實現(xiàn)模型可視化

2023-02-24 10:07 作者:曠視天元MegEngine  | 我要投稿

近期社區(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 格式

  1. Netron 需要是 6.0.0 及以上版本,為了保證最優(yōu)體驗效果,建議使用最新版本;

  2. 保存 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;

  3. 因為 tm 格式的模型有很多子 module ,為了讓大家更好的了解他們之間的鏈接關系,所以在做可視化展示時,各子圖是全部展開的,以一張圖來完整展示模型結構。

C++ 格式

  1. Netron 使用 6.5.3 及以上版本,為了保證最優(yōu)體驗效果,建議使用最新版本;

  2. 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 MegEngineerhttps://www.megengine.org.cn/community-AMGE,榮譽證書、定制禮品享不停


MegEngine 使用小技巧:使用 Netron 實現(xiàn)模型可視化的評論 (共 條)

分享到微博請遵守國家法律
泊头市| 鸡西市| 天津市| 玉环县| 汝阳县| 安多县| 阜新| 东至县| 独山县| 吴桥县| 亳州市| 汉中市| 荔浦县| 都兰县| 鄂伦春自治旗| 江陵县| 琼海市| 漳浦县| 盐山县| 西充县| 金湖县| 临海市| 湾仔区| 和硕县| 襄垣县| 鄄城县| 长泰县| 定西市| 利辛县| 神池县| 栾川县| 文登市| 建始县| 嵊泗县| 泰州市| 鄂尔多斯市| 襄城县| 苏尼特右旗| 商丘市| 闻喜县| 东乌|