把模型文件拖进来,看看网络长什么样
Netron · 开源项目,2017 年起持续维护
学生训完一个模型、拿到一个文件,里面到底是什么?把文件拖进这个页面,整张计算图就画出来了:每一层是什么算子、张量形状多大、参数有多少,点开任意一层右侧逐项列出。全部在浏览器本地解析,模型不上传。
交互实验(可直接投屏)
原作是一个完整的网页应用,下面直接嵌入;屏幕小或要投影时,点「全屏打开实验」更清楚。
动手三步
- 首屏点「打开模型文件」,或直接把模型文件拖到页面上
- 左边是整张计算图,从上往下就是数据流向;点某一层的标题条,右侧列出类型、名称以及各项属性与输入输出
- 点工具栏看模型属性:模型格式、生成工具、运行时
- 用快捷键切换显示算子属性、隐藏权重、改为横向布局
- 菜单里「导出为图片」可以把整张结构图存下来,直接贴进课件或论文
- 手边没有模型文件时,可以用地址栏参数加载本站另一件展品里的模型
课堂怎么用
- 让学生把自己训练出来的模型拖进来,当堂数一数有多少层、总参数量多大,再和论文里的示意图对照
- 讲模型交换格式时,同一个模型分别导出两种格式拖进来,比较算子被怎样拆开
- 答辩前让学生用导出功能生成结构图,替代手画的方框图
- 接 B-09 预训练模型三维可视化:那件看模型跑起来的样子,这件看它的静态结构
思考题
- 卷积层的卷积核形状里四个数字分别代表什么?这一层的参数量怎么算出来?
- 从展平到全连接,张量形状为什么变成一维?那个数是怎么来的?
- 同一个模型导出成不同格式后,图里的算子数量为什么会变?
已知限制
- 镜像里没有自带示例模型(原仓库本就不含模型文件),课堂上需老师自备一个模型文件,或用地址栏参数加载本站另一件展品里的模型
- 菜单里的反馈入口指向站外,课堂断网时打不开
- 计算图里的算子名、张量名、形状,以及模型自带的属性名是模型自身的数据,按原样显示不翻译
- 原作首访会发一次归属地查询、并用它决定是否弹出同意框,断网时同意框会卡住导致打不开模型;本镜像已整段移除,遥测也已加硬闸
原作 Netron(Lutz Roeder,开源项目,2017 年起持续维护),MIT 许可(原文),界面已中文化。