项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 11314
使用教程
ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。
环境要求
- Python 环境(wandb 为 Python 库,通过 pip 安装)
- 已安装 pip 包管理工具
- 需要注册 W&B 账号(README 建议 sign up for a W&B account)
安装与启动步骤
-
1安装 wandb 库
使用 pip 安装 W&B 官方 Python 包,README 明确给出的安装命令。建议在虚拟环境中安装,避免污染全局环境。
pip install wandb -
2登录 W&B 账号
首次使用需在终端登录以关联账号;README 只给出注册入口,登录命令为常规 CLI 用法,如不确定可改用环境变量方式。
wandb login -
3编写训练脚本
在训练代码中设置 project 名称,用 wandb.init 开启一次 run,并用 run.log 记录 accuracy、loss 等指标。
-
4运行脚本查看结果
执行训练脚本后,访问 wandb.ai/home 查看 Runs 列中自动命名的 run 及 accuracy、loss 随训练步数的变化曲线。
python train.py
关键配置
| 配置项 | 必填 | 说明 | 示例 |
|---|---|---|---|
project | 是 | 指定 run 记录到的项目名称 | my-awesome-project |
config | 否 | 传入训练配置字典,随 run 一起记录超参数 | {"learning_rate": 0.001} |
如何确认成功
脚本运行无报错,且在 wandb.ai/home 的项目 Runs 列中能看到本次 run 及记录的 accuracy、loss 曲线。
常见问题
Q:提示未登录或缺少 API Key 怎么办?
A:先在 https://wandb.ai 注册账号,再运行 wandb login 按提示粘贴 API Key;也可设置 WANDB_API_KEY 环境变量后运行。
Q:笔记本(Notebook)里怎么用?
A:README 指出 notebook 中可直接 run = wandb.init(),并在结束时手动调用 run.finish(),而不用 with 块。
Q:run 失败状态是怎么标记的?
A:使用 with 语法时,代码块正常退出会将 run 标记为 finished,若块内抛出异常则标记为 failed。
Q:一定要联网才能用吗?
A:README 未说明离线模式,默认记录会同步到 W&B 云端,需联网并登录账号才能在 wandb.ai/home 查看结果。
注意事项
- wandb 是 Python 客户端库而非独立服务端,无需 Docker 或本地部署服务。
- README 只给出 pip install wandb 这一条安装命令,其他方式(如 conda)未在节选中说明。
- 用 with wandb.init(...) 包裹训练代码可自动结束 run,推荐加上以避免 run 悬挂。
核心亮点
- 实验跟踪与可视化一体化,支持指标、超参数、模型及数据版本管理,大幅提升团队协作效率
- 提供从实验到生产的全流程管理,涵盖模型注册、部署监控,降低MLOps落地门槛
- 集成主流深度学习框架(PyTorch、TensorFlow等),API简洁,上手成本低
不足之处
- 免费版功能受限,高级特性(如团队协作、私有云)需付费,对个人开发者不友好
- 依赖云端服务,数据安全与隐私存在顾虑,自托管部署复杂度较高
适用场景
- 深度学习模型训练过程中的实验对比与超参数调优
- 团队协作开发,统一管理实验记录与模型版本
- 模型从实验到生产的全生命周期监控与治理
替代项目
MLflow、Neptune.ai、Comet.ml
同类项目推荐
Awesome-MLLM-Tuning
开源
多模态大模型微调不丢本事的论文与思路合集
Multimodal Large Language Model (MLLM) Tuning Survey: Keeping Yourself is Important ···
ts-icl
开源
零样本搞定时间序列预测与插补,无需重新训练。
TS-ICL: a Time-Indexed Foundation Model for Time Series Forecasting & Imputation via···
q-diffusion
开源
把扩散模型压到低比特,省内存还提速,生成质量不掉线。
[ICCV 2023] Q-Diffusion: Quantizing Diffusion Models.
perfusion-pytorch
开源
用少量样本给文本生成模型打补丁,实现精准图像编辑
Implementation of Key-Locked Rank One Editing, from Nvidia AI
llm-foundry
开源
开箱即用的生产级 LLM 训练框架,快速搞定预训练与微调
LLM training code for Databricks foundation models
dashAI
开源
可视化训练评估部署,一条龙搞定 AI 模型
dashAI: an interactive platform for training, evaluating and deploying AI models
Agent-FLAN
开源
让大模型学会用工具,微调数据与方法一键搞定
[ACL2024 Findings] Agent-FLAN: Designing Data and Methods of Effective Agent Tuning ···
Adam-mini
开源
用更少学习率,训更快更大模型,省显存提性能。
Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406···
hyperdreambooth
开源
秒级生成专属人像,无需漫长训练
Implementation of HyperDreamBooth: HyperNetworks for Fast Personalization of Text-to···
