AMD, in collaboration with the ONNX Community and the Linux Foundation, has released an open-source tool to help developers explore and understand machine learning models. Digest is a powerful model analysis tool designed to help you extract valuable insights from your machine learning models. With Digest, you can easily organize, report on, and optimize your models, making your workflow more efficient and effective.
Why Use Digest AI:
- Model Analysis: The tool performs model analysis, providing details such as parameters and FLOPs, along with advanced insights like operation histograms and model similarity graphs.
- Platform Compatibility: It works on both Windows and Linux platforms, offering flexibility for various development environments.
- Reporting: Digest AI enables you to save your model analysis by generating reports that summarize the analysis, including all nodes, parameters, FLOPs, attributes, and inputs and outputs of the model. Additionally, you can save images of the heatmap, which shows the model's similarity to other public models, and histograms of operation counts.
- Multi-Model Analysis: It supports multi-model analysis, enabling you to analyze multiple models simultaneously and view a global summary.
- Hugging Face Integration: The tool includes a beta module for downloading and analyzing ONNX models directly from the Hugging Face hub.
- Dynamic Input Handling: Digest AI can manage models with dynamic input shapes, enabling you to specify static dimensions for more accurate analysis.
- API Support: It comes with an API for ingesting, modifying, and analyzing machine learning models.
Getting Started with Digest AI :
With just one click, Digest AI enables you to extract insights from your models and works on both Windows and Linux platforms. To get started, you can install Digest AI from the source code available on the GitHub repository at github.com/onnx/digest. Once you have completed the installation, run digest from your command line to launch the application. In the GitHub repository, there are also instructions on how to build an executable, which can then launch the application. To get in touch with the Digest AI developers, reach out on GitHub or at digestai@amd.com.
Digest AI is a powerful and user-friendly tool that can significantly enhance your machine learning development. By providing detailed model analysis, multi-model support, and integration with the Hugging Face hub, Digest AI helps you analyze your models. To learn more about Digest AI, read the full article Introducing Digest AI: A Model Ingestion and Analysis Tool