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 工程师。
  • 学习新框架并寻找运行日志记录具体示例的学生。
  • 需要共享仪表板以比较超参数搜索、模型性能和资源使用情况的团队。

如何开始

  1. 安装核心库:
    pip install wandb
    wandb login   # 粘贴从 wandb.com 获取的 API 密钥
    
  2. 选择与您技术栈匹配的示例(例如 examples/pytorch/train.pycolabs/keras.ipynb)。
  3. 运行脚本或笔记本;新的运行将出现在您的 W&B 仪表板中,您可以探索记录的指标、配置和制品。

您将看到的内容 – README 包含 W&B 仪表板的截图和一个简短视频导览链接,让您在深入使用前对 UI 有直观感受。


以上所有信息均直接来自仓库的 README;未推断任何额外功能。

相关

  • 项目
  • 项目
  • Dispatch
  • 项目
  • 项目