K-文化的一切——从回归到 K-美妆,发送到您的邮箱订阅邮件

METAL MEDIA

google-ai-edge/model-explorer

1,532JavaScriptApache-2.0

一款把深度学习模型内部结构画出来看的工具

Model Explorer 把深度学习模型的运算图以分层结构可视化出来。用户可以展开或折叠各层,高亮输入输出运算,并在节点上叠加显示元数据。它支持 TFLite、TF、TFJS、MLIR 和 PyTorch(Exported Program)格式,并提供扩展框架以支持更多格式。

它做什么

  1. 把模型运算组织成可展开折叠的嵌套层级,方便探索图结构
  2. 提供输入输出运算高亮、节点元数据叠加显示、交互式弹窗查看层、搜索、识别相同层、GPU加速渲染等功能
  3. 执行 pip install ai-edge-model-explorer 后运行 model-explorer 命令即可使用,也可以在 Hugging Face 上传模型直接体验
  4. 核心可视化组件也以 ai-edge-model-explorer-visualizer 这个 npm 包发布
  5. 社区已经为 ONNX、VGF、TOSA 格式开发了独立的适配器仓库

为什么重要

模型结构越来越复杂时,单靠阅读代码很难搞清楚各个运算是怎么连接的,这个工具让人可以直观地查看结构并进行调试。它的适配器扩展框架也让开发者可以自行添加对 TFLite、PyTorch 之外格式的支持。

本仓库术语

  • TFLite · 用于在移动和嵌入式设备上运行的轻量级 TensorFlow 模型格式
  • MLIR · 多个编译器共用的一种中间表示格式
  • Exported Program · PyTorch 模型导出后得到的图结构表示
  • 适配器(Adapter) · 把特定模型格式转换成该工具可读取形式的扩展模块

仓库简介(英文)

A modern model graph visualizer and debugger

所属 · google-deepmind

在 GitHub 打开

热门仓库

全部仓库 →

METAL MEDIA 最新报道