HyperAIHyperAI

Command Palette

Search for a command to run...

Trackio : une bibliothèque légère pour suivre et partager vos expériences de machine learning

Trackio est une bibliothèque d'analyse d'expériences légère développée par Hugging Face, conçue pour aider les chercheurs en intelligence artificielle à suivre les métriques, les paramètres et les hyperparamètres lors de l'entraînement de modèles. Elle a été créée pour répondre aux limites des outils existants, qui peuvent être coûteux, complexes à configurer ou manquer de flexibilité. Trackio permet de partager facilement les progrès d'entraînement via des iframes, de standardiser les données avec une transparence accrue, d'accéder facilement aux données sans dépendre d'API propriétaires, et de tester rapidement de nouvelles fonctionnalités sans affecter les performances. Trackio est une bibliothèque open source en Python qui permet de suivre les métriques et de les visualiser via un tableau de bord local basé sur Gradio. Elle peut être synchronisée avec Hugging Face Spaces, ce qui permet de partager les résultats via un simple lien URL. Les espaces peuvent être privés ou publics, offrant ainsi une flexibilité totale. Les données sont stockées dans une base de données Sqlite éphémère, mais Trackio les convertit automatiquement en format Parquet et les sauvegarde dans un dataset Hugging Face toutes les 5 minutes, pour une meilleure accessibilité. Trackio s'intègre nativement aux bibliothèques Hugging Face comme Transformers et Accelerate. Avec Transformers, il suffit de configurer l'argument report_to="trackio" pour commencer à suivre les métriques. Avec Accelerate, il est possible de lancer le suivi avec une simple configuration log_with="trackio". Aucun paramétrage supplémentaire n'est nécessaire, ce qui rend son utilisation rapide et intuitive. Les principes de conception de Trackio visent à rester léger, simple et accessible, tout en offrant des fonctionnalités essentielles. Bien qu'elle soit encore en version bêta, elle ne dispose pas encore de certaines fonctionnalités comme la gestion des artefacts ou des visualisations avancées. Cependant, Hugging Face encourage la communauté à contribuer et à proposer des améliorations via GitHub. Cette approche collaborative permet de construire un outil adapté aux besoins de tous les utilisateurs. Trackio semble être une solution prometteuse pour les chercheurs qui souhaitent suivre leurs expériences de manière simple, transparente et gratuite.

Liens associés

Trackio : une bibliothèque légère pour suivre et partager vos expériences de machine learning | Articles tendance | HyperAI