Skip to content

cosmopax/model-explorer

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

95 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Model Explorer PyPI version

Model Explorer offers an intuitive and hierarchical visualization of model graphs. It organizes model operations into nested layers, enabling users to dynamically expand or collapse these layers. It also provides a range of features to facilitate model exploration and debugging, including the ability to highlight input and output operations, overlay metadata on nodes, display layers in interactive pop-ups, perform searches, show identical layers, GPU-accelerated graph rendering, among others. It currently supports TFLite, TF, TFJS, MLIR, and PyTorch (Exported Program) model format, and provides an extension framework for developers to easily add support for additional formats.

Home page screenshot


To start using Model Explorer, run:

$ pip install ai-edge-model-explorer
$ model-explorer

Please check out our Wiki for more details:

We invite you to participate in our future research studies on Model Explorer. Sign up here.

Contributions

Developers are invited to create custom adapters to extend support for additional model formats. Please refer to our Develop Adapter Extension guide. We will gladly accept pull requests to add links to the GitHub repository and PyPI package of your custom adapter to our README.

About

A modern model graph visualizer and debugger

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • JavaScript 89.5%
  • C++ 6.5%
  • Python 3.2%
  • Starlark 0.4%
  • Jupyter Notebook 0.3%
  • Shell 0.1%