wandb/examples
Example deep learning projects that use wandb's features.
wandb/examples – 使用 Weights & Biases 的实验跟踪即用代码
是什么 – 本仓库是一组简短的脚本和 Colab 笔记本,展示了如何将 Weights & Biases (W&B) 库集成到典型的机器学习工作流中。示例覆盖了多个流行框架(PyTorch、TensorFlow/Keras、Hugging Face Transformers、PyTorch Lightning、XGBoost、scikit‑learn 等),并演示了 W&B 的核心功能:记录指标、保存超参数、追踪数据集、可视化运行、管理模型制品。
为何重要 – W&B 是一个用于实验跟踪、模型版本控制和协作报告的平台。通过查看这些示例,您可以看到,仅用少量代码即可实现:
- 启动一个运行(
wandb.init)并关联项目名称。 - 记录配置值(
wandb.config)。 - 记录标量指标(
wandb.log)。 - 自动捕获模型梯度、权重和拓扑结构(
run.watch)。 - 使用框架特定的辅助工具(例如
wandb.keras.WandbMetricsLogger、XGBoost 的WandbCallback、PyTorch Lightning 的WandbLogger)。 - 通过 W&B Artifacts 存储和检索数据集或模型检查点。
- 清晰地结束一个运行(
wandb.finish)。
仓库的主要部分
examples/– 每个框架的即用 Python 脚本,展示记录训练循环所需的最少代码。colabs/– Google‑Colab 笔记本,无需任何本地设置即可在浏览器中尝试集成。- README 中的简短 「快速入门」 部分,逐步指导您安装
wandb包、登录账号,并运行一个简单的示例,记录十步内的损失值。
典型用户
- 希望为现有项目添加实验跟踪功能的数据科学家和 ML 工程师。
- 学习新框架并寻找运行日志记录具体示例的学生。
- 需要共享仪表板以比较超参数搜索、模型性能和资源使用情况的团队。
如何开始
- 安装核心库:
pip install wandb wandb login # 粘贴从 wandb.com 获取的 API 密钥 - 选择与您技术栈匹配的示例(例如
examples/pytorch/train.py或colabs/keras.ipynb)。 - 运行脚本或笔记本;新的运行将出现在您的 W&B 仪表板中,您可以探索记录的指标、配置和制品。
您将看到的内容 – README 包含 W&B 仪表板的截图和一个简短视频导览链接,让您在深入使用前对 UI 有直观感受。
以上所有信息均直接来自仓库的 README;未推断任何额外功能。
相关
- 项目
- 项目
- Dispatch
- 项目
- 项目