SwanHubX/SwanLab

⚡️SwanLab - an open-source, modern-design AI training tracking and visualization tool. Supports Cloud / Self-hosted use. Integrated with PyTorch / Transformers / verl / LLaMA Factory / ms-swift / Ultralytics / MMEngine / Keras etc.

解决的问题

SwanLab 是专为模型训练团队设计的 AI 训练分析与指标观测平台。它通过提供可视化、自动日志记录、超参数记录和实验比较工具，解决了追踪复杂训练过程的问题，帮助研究人员识别训练问题并加速模型迭代。

工作原理

用户将轻量级 SDK 集成到其机器学习流水线中，以记录标量、图像、音频、文本、视频和 3D 点云。随后，平台通过集中式仪表板将这些数据可视化。它支持用于远程监控的云端版本和用于离线环境的自托管社区版本。它还与 PyTorch、HuggingFace Transformers 和 LLaMA Factory 等 30 多个主流框架集成。

适用对象

专为需要监控训练进度、通过比较多个实验来寻找最优超参数以及与团队成员实时协作的 AI 研究人员和模型训练团队而构建。

亮点

  • 全面的指标追踪：支持包括标量、3D 点云、生化分子和自定义 ECharts 在内的多种数据类型。
  • 广泛的框架集成：与包括 PyTorch Lightning、Ultralytics 和 XGBoost 在内的 30 多个框架无缝集成。
  • 广泛的硬件监控：对包括 NVIDIA GPUs、Ascend NPUs、AMD ROCm 及其他各种专用 AI 加速器在内的广泛硬件进行实时系统级监控。
  • 协作工具：具备项目共享、团队协作以及通过 Slack、Discord 和 Lark 进行通知的插件扩展功能。
  • 灵活的部署：提供托管云服务和通过 Docker 及 Kubernetes 实现的自托管选项。