Sobre o projeto
Aim é uma ferramenta de código aberto e auto-hospedada para rastreamento de experimentos em aprendizado de máquina, projetada para lidar com um grande número de execuções de treinamento. Ela registra execuções de treinamento e metadados de IA, fornece uma interface para comparar e observar essas execuções, e expõe uma API para consultas programáticas.
Capacidades principais descritas no README:
- Registro de experimentos de ML e metadados arbitrários, com integrações para frameworks populares de ML.
- Uma interface para visualizar e comparar metadados, incluindo agrupamento, agregação e consultas via expressões Python.
- Rastreamento de informações do sistema e uso de recursos, alertas em tempo real sobre o progresso do treinamento e notificações configuráveis.
- Organização de execuções por tags e experimentos, informações detalhadas de execução para depuração e um painel centralizado.
- Conversores integrados para migrar logs de TensorBoard, MLflow e Weights & Biases; conversões personalizadas possíveis via SDK.
- Integrações listadas incluem PyTorch Ignite, PyTorch Lightning, Hugging Face, Keras/tf.Keras, Keras Tuner, XGBoost, CatBoost, LightGBM, fastai, MXNet e Optuna.
- Tipos de objetos rastreáveis suportados incluem métricas, imagens, texto, áudio e distribuições.
Início rápido: instale via pip, crie uma execução (Run), registre parâmetros e métricas com run.track, e depois inicie a interface com `aim up`. O projeto também aponta para um ecossistema de ferramentas baseadas em Aim, como aimlflow e Aim-spaCy, e oferece demonstrações ao vivo para experimentos de tradução automática, GAN, FastSpeech 2 e MNIST.
Comments
0 people shared their preference · Deer Point appears after 10 participants
Sign in to join the discussion.