AMD, en colaboración con la comunidad ONNX y la Fundación Linux, ha lanzado una herramienta de código abierto para ayudar a los desarrolladores a explorar y comprender los modelos de aprendizaje automático.
Digest es una potente herramienta de análisis de modelos diseñada para ayudarlo a extraer información valiosa de sus modelos de aprendizaje automático. Con Digest, puede organizar, informar y optimizar fácilmente sus modelos, lo que hace que su flujo de trabajo sea más eficiente y eficaz.
Por qué usar Digest AI:
Análisis de modelos: la herramienta realiza análisis de modelos y proporciona detalles como parámetros y FLOP, junto con información avanzada como histogramas de operaciones y gráficos de similitud de modelos.
Compatibilidad de plataformas: funciona tanto en plataformas Windows como Linux, lo que ofrece flexibilidad para varios entornos de desarrollo.
Informes: Digest AI le permite guardar su análisis de modelos al generar informes que resumen el análisis, incluidos todos los nodos, parámetros, FLOP, atributos y entradas y salidas del modelo. Además, puede guardar imágenes del mapa de calor, que muestra la similitud del modelo con otros modelos públicos, e histogramas de recuentos de operaciones.
Análisis multimodelo: admite el análisis multimodelo, lo que le permite analizar varios modelos simultáneamente y ver un resumen global.
Integración de Hugging Face: la herramienta incluye un módulo beta para descargar y analizar modelos ONNX directamente desde el centro Hugging Face.
Manejo de entrada dinámica: Digest AI puede administrar modelos con formas de entrada dinámicas, lo que le permite especificar dimensiones estáticas para un análisis más preciso.
Compatibilidad con API: viene con una API para ingerir, modificar y analizar modelos de aprendizaje automático.
Primeros pasos con Digest AI:
Con solo un clic, Digest AI le permite extraer información de sus modelos y funciona en plataformas Windows y Linux. Para comenzar, puede instalar Digest AI desde el código fuente disponible en el repositorio de GitHub en github.com/onnx/digest. Una vez que haya completado la instalación, ejecute digest desde su línea de comandos para iniciar la aplicación. En el repositorio de GitHub, también hay instrucciones sobre cómo crear un ejecutable, que luego puede iniciar la aplicación. Para ponerse en contacto con los desarrolladores de Digest AI, comuníquese a través de GitHub o a través de digestai@amd.com.
Digest AI es una herramienta potente y fácil de usar que puede mejorar significativamente el desarrollo de aprendizaje automático. Al proporcionar un análisis detallado de modelos, compatibilidad con múltiples modelos e integración con el centro Hugging Face, Digest AI lo ayuda a analizar sus modelos. Para obtener más información sobre Digest AI, lea el artículo completo Presentación de Digest AI: una herramienta de análisis e ingesta de modelos