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 實現的自託管選項。