SwanLab

一个开源、现代化设计的深度学习训练跟踪与可视化工具
同时支持云端/离线使用，适配30+主流框架，与你的实验代码轻松集成

🔥SwanLab 在线版 · 📃 文档 · 报告问题 · 建议反馈 · 更新日志 · 基线社区

中文 / English / 日本語 / Русский

👋 加入我们的微信群

目录

• 🌟 最近更新
• 👋🏻 什么是SwanLab
• 📃 在线演示
• 🏁 快速开始
• 💻 自托管
• 🚗 框架集成
• 🔌 插件与API
• 🎮 硬件记录
• 🆚 与熟悉的工具的比较
• 👥 社区
• 📃 协议

🌟 最近更新

• 2025.06.11：📊支持 swanlab.echarts.table 数据类型，支持纯文本图表展示；支持对分组进行拉伸交互，以增大同时显示的图表数量；表格视图增加 指标最大/最小值 选项；

• 2025.06.08：♻️支持在本地存储完整的实验日志文件，通过 swanlab sync 上传本地日志文件到云端/私有化部署端；硬件监控支持海光DCU；

• 2025.06.01：🏸支持图表自由拖拽；支持ECharts自定义图表，增加包括柱状图、饼状图、直方图在内的20+图表类型；硬件监控支持沐曦GPU；集成 PaddleNLP 框架；

• 2025.05.25：日志支持记录标准错误流，PyTorch Lightning等框架的打印信息可以被更好地记录；硬件监控支持摩尔线程；新增运行命令记录安全防护功能，API Key将被自动隐藏；

• 2025.05.14：支持实验Tag；支持折线图Log Scale；支持分组拖拽；大幅度优化了大量指标上传的体验；增加swanlab.OpenApi开放接口；

• 2025.05.09：支持折线图创建；配置图表功能增加数据源选择功能，支持单张图表显示不同的指标；支持生成训练项目GitHub徽章；

• 2025.04.23：支持折线图编辑，支持自由配置图表的X、Y轴数据范围和标题样式；图表搜索支持正则表达式；支持昆仑芯XPU的硬件检测与监控；

• 2025.04.11：支持折线图局部区域选取；支持全局选择仪表盘折线图的step范围；支持一键隐藏全部图表；

• 2025.04.08：支持swanlab.Molecule数据类型，支持记录与可视化生物化学分子数据；支持保存表格视图中的排序、筛选、列顺序变化状态；

👋🏻 什么是SwanLab

SwanLab 是一款开源、轻量的 AI 模型训练跟踪与可视化工具，提供了一个跟踪、记录、比较、和协作实验的平台。

SwanLab 面向人工智能研究者，设计了友好的Python API 和漂亮的UI界面，并提供训练可视化、自动日志记录、超参数记录、实验对比、多人协同等功能。在SwanLab上，研究者能基于直观的可视化图表发现训练问题，对比多个实验找到研究灵感，并通过在线网页的分享与基于组织的多人协同训练，打破团队沟通的壁垒，提高组织训练效率。

https://github.com/user-attachments/assets/7965fec4-c8b0-4956-803d-dbf177b44f54

以下是其核心特性列表：

1. 📊 实验指标与超参数跟踪: 极简的代码嵌入您的机器学习 pipeline，跟踪记录训练关键指标

• ☁️ 支持云端使用（类似Weights & Biases），随时随地查看训练进展。手机看实验的方法

• 📝 支持超参数记录、指标总结、表格分析

• 🌸 可视化训练过程: 通过UI界面对实验跟踪数据进行可视化，可以让训练师直观地看到实验每一步的结果，分析指标走势，判断哪些变化导致了模型效果的提升，从而整体性地提升模型迭代效率。

• 支持的元数据类型：标量指标、图像、音频、文本、3D点云、生物化学分子、Echarts自定义图表...

• 支持的图表类型：折线图、媒体图（图像、音频、文本）、3D点云、生物化学分子、柱状图、散点图、箱线图、热力图、饼状图、雷达图...

• 后台自动记录：日志logging、硬件环境、Git 仓库、Python 环境、Python 库列表、项目运行目录

2. ⚡️ 全面的框架集成: PyTorch、🤗HuggingFace Transformers、PyTorch Lightning、🦙LLaMA Factory、MMDetection、Ultralytics、PaddleDetetion、LightGBM、XGBoost、Keras、Tensorboard、Weights&Biases、OpenAI、Swift、XTuner、Stable Baseline3、Hydra 在内的 30+ 框架

3. 💻 硬件监控: 支持实时记录与监控CPU、NPU（昇腾Ascend）、GPU（英伟达Nvidia）、MLU（寒武纪Cambricon）、XLU（昆仑芯Kunlunxin）内存的系统级硬件指标

4. 📦 实验管理: 通过专为训练场景设计的集中式仪表板，通过整体视图速览全局，快速管理多个项目与实验

5. 🆚 比较结果: 通过在线表格与对比图表比较不同实验的超参数和结果，挖掘迭代灵感

6. 👥 在线协作: 您可以与团队进行协作式训练，支持将实验实时同步在一个项目下，您可以在线查看团队的训练记录，基于结果发表看法与建议

7. ✉️ 分享结果: 复制和发送持久的 URL 来共享每个实验，方便地发送给伙伴，或嵌入到在线笔记中

8. 💻 支持自托管: 支持离线环境使用，自托管的社区版同样可以查看仪表盘与管理实验，使用攻略

9. 🔌 插件拓展: 支持通过插件拓展SwanLab的使用场景，比如 飞书通知、Slack通知、CSV记录器等

[!IMPORTANT]

收藏项目，你将从 GitHub 上无延迟地接收所有发布通知～ ⭐️

📃 在线演示

来看看 SwanLab 的在线演示：

ResNet50 猫狗分类	Yolov8-COCO128 目标检测
跟踪一个简单的 ResNet50 模型在猫狗数据集上训练的图像分类任务。	使用 Yolov8 在 COCO128 数据集上进行目标检测任务，跟踪训练超参数和指标。

Qwen2 指令微调	LSTM Google 股票预测
跟踪 Qwen2 大语言模型的指令微调训练，完成简单的指令遵循。	使用简单的 LSTM 模型在 Google 股价数据集上训练，实现对未来股价的预测。

ResNeXt101 音频分类	Qwen2-VL COCO数据集微调
从ResNet到ResNeXt在音频分类任务上的渐进式实验过程	基于Qwen2-VL多模态大模型，在COCO2014数据集上进行Lora微调。

EasyR1 多模态LLM RL训练	Qwen2.5-0.5B GRPO训练
使用EasyR1框架进行多模态LLM RL训练	基于Qwen2.5-0.5B模型在GSM8k数据集上进行GRPO训练

更多案例

🏁 快速开始

1.安装

pip install swanlab

# 方式一
git clone https://github.com/SwanHubX/SwanLab.git
pip install -e .

# 方式二
pip install git+https://github.com/SwanHubX/SwanLab.git

pip install 'swanlab[dashboard]'

2.登录并获取 API Key

1. 免费注册账号

2. 登录账号，在用户设置 > API Key 里复制您的 API Key

3. 打开终端，输入：

swanlab login

出现提示时，输入您的 API Key，按下回车，完成登陆。

3.将 SwanLab 与你的代码集成

import swanlab

# 初始化一个新的swanlab实验
swanlab.init(
    project="my-first-ml",
    config={'learning-rate': 0.003},
)

# 记录指标
for i in range(10):
    swanlab.log({"loss": i, "acc": i})

大功告成！前往SwanLab查看你的第一个 SwanLab 实验。

💻 自托管

自托管社区版支持离线查看 SwanLab 仪表盘。

1. 使用Docker部署自托管版本

详情请参考：文档

git clone https://github.com/SwanHubX/self-hosted.git
cd self-hosted/docker

中国地区快速安装：

./install.sh

从DockerHub拉取镜像安装：

./install-dockerhub.sh

2. 将实验指定到自托管服务

登录到自托管服务：

swanlab login --host http://localhost:8000

完成登录后，即可将实验记录到自托管服务。

🚗 框架集成

将你最喜欢的框架与 SwanLab 结合使用！
下面是我们已集成的框架列表，欢迎提交 Issue 来反馈你想要集成的框架。

基础框架

• PyTorch
• MindSpore
• Keras

专有/微调框架

• PyTorch Lightning
• HuggingFace Transformers
• LLaMA Factory
• Modelscope Swift
• DiffSynth Studio
• Sentence Transformers
• PaddleNLP
• OpenMind
• Torchtune
• XTuner
• MMEngine
• FastAI
• LightGBM
• XGBoost

评估框架

• EvalScope

计算机视觉

• Ultralytics
• MMDetection
• MMSegmentation
• PaddleDetection
• PaddleYOLO

强化学习

• Stable Baseline3
• veRL
• HuggingFace trl
• EasyR1

其他框架：

• Tensorboard
• Weights&Biases
• MLFlow
• HuggingFace Accelerate
• Unsloth
• Hydra
• Omegaconf
• OpenAI
• ZhipuAI

更多集成

🔌 插件与API

欢迎通过插件来拓展SwanLab的功能，增强你的实验管理体验！

• 自定义你的插件
• 邮件通知
• 飞书通知
• 钉钉通知
• 企业微信通知
• Discord通知
• Slack通知
• CSV记录器

开放接口：

• OpenAPI

🎮 硬件记录

SwanLab会对AI训练过程中所使用的硬件信息和资源使用情况进行记录，下面是支持情况表格：

如果你希望记录其他硬件，欢迎提交Issue与PR！

🆚 与熟悉的工具的比较

Tensorboard vs SwanLab

• ☁️ 支持在线使用： 通过 SwanLab 可以方便地将训练实验在云端在线同步与保存，便于远程查看训练进展、管理历史项目、分享实验链接、发送实时消息通知、多端看实验等。而 Tensorboard 是一个离线的实验跟踪工具。

• 👥 多人协作： 在进行多人、跨团队的机器学习协作时，通过 SwanLab 可以轻松管理多人的训练项目、分享实验链接、跨空间交流讨论。而 Tensorboard 主要为个人设计，难以进行多人协作和分享实验。

• 💻 持久、集中的仪表板： 无论你在何处训练模型，无论是在本地计算机上、在实验室集群还是在公有云的 GPU 实例中，你的结果都会记录到同一个集中式仪表板中。而使用 TensorBoard 需要花费时间从不同的机器复制和管理 TFEvent 文件。

• 💪 更强大的表格： 通过 SwanLab 表格可以查看、搜索、过滤来自不同实验的结果，可以轻松查看数千个模型版本并找到适合不同任务的最佳性能模型。 TensorBoard 不适用于大型项目。

Weights and Biases vs SwanLab

• Weights and Biases 是一个必须联网使用的闭源 MLOps 平台

• SwanLab 不仅支持联网使用，也支持开源、免费、自托管的版本

👥 社区

周边仓库

• SwanLab-Docs：官方文档仓库
• SwanLab-Dashboard：离线看板仓库，存放了由swanlab watch打开的轻量离线看板的web代码
• self-hosted：私有化部署脚本仓库

社区与支持

• GitHub Issues：使用 SwanLab 时遇到的错误和问题
• 电子邮件支持：反馈关于使用 SwanLab 的问题
• 微信交流群：交流使用 SwanLab 的问题、分享最新的 AI 技术

SwanLab README 徽章

如果你喜欢在工作中使用 SwanLab，请将 SwanLab 徽章添加到你的 README 中：

、

[![](https://raw.githubusercontent.com/SwanHubX/assets/main/badge2.svg)](your experiment url)
[![](https://raw.githubusercontent.com/SwanHubX/assets/main/badge1.svg)](your experiment url)

更多设计素材：assets

在论文中引用 SwanLab

如果您发现 SwanLab 对您的研究之旅有帮助，请考虑以下列格式引用：

@software{Zeyilin_SwanLab_2023,
  author = {Zeyi Lin, Shaohong Chen, Kang Li, Qiushan Jiang, Zirui Cai,  Kaifang Ji and {The SwanLab team}},
  doi = {10.5281/zenodo.11100550},
  license = {Apache-2.0},
  title = {{SwanLab}},
  url = {https://github.com/swanhubx/swanlab},
  year = {2023}
}

为 SwanLab 做出贡献

考虑为 SwanLab 做出贡献吗？首先，请花点时间阅读 贡献指南。

同时，我们非常欢迎通过社交媒体、活动和会议的分享来支持 SwanLab，衷心感谢！

Contributors

📃 协议

本仓库遵循 Apache 2.0 License 开源协议

Star History