使用Weights & Biases 简化机器学习工作流程。跟踪、可视化和协作实验,实现更快、可重复的人工智能开发。
Weights & Biases (Weights & Biases ,通常缩写为 W&B 或 WandB)是一个著名的开发人员优先平台,旨在 简化 机器学习操作(MLOps)的著名平台。它是机器学习工程师和数据科学家的集中记录系统,使他们能够 track 实验、可视化模型性能并管理数据集。在复杂的机器学习(ML)生命周期中 在复杂的机器学习(ML)生命周期中,track 每个 在复杂的机器学习(ML)生命周期中,跟踪每一个配置变化和结果都很困难。 解决了这个问题。这样,团队就可以比较不同的训练运行,确保可重复性,并 在从学术研究到企业规模的项目中进行有效协作。 计算机视觉 (CV)部署等项目上进行有效协作。
要了解该工具,必须将其与它所命名的基本 神经网络 (NN)概念的区别。
参数weights and biases)是在训练过程中通过随机梯度下降(SGD)优化的数学成分。 随机梯度下降法(SGD)进行优化、 平台Weights & Biases)是用于观察和分析优化过程的工具。
W&B 平台提供了一整套工具,以应对人工智能(AI)开发过程中的特定挑战。 人工智能 (AI) 工作流程中的具体挑战。
Weights & Biases 》广泛应用于需要严格模型验证和协作开发的行业。
将 "Weights & Biases 与现代框架(如 PyTorch等现代框架或Ultralytics 等特定库的整合是 简单明了。该库可自动检测 W&B 的存在并记录关键指标,无需大量的 模板代码。
下面的示例演示了如何训练YOLO11 模型,同时将运行情况自动记录到 Weights & Biases 中。
偏差。确保您已运行 pip install wandb 并通过 wandb login 在
运行脚本前,请在终端中
from ultralytics import YOLO
# Initialize the YOLO11 model using pre-trained weights
model = YOLO("yolo11n.pt")
# Train the model on the COCO8 dataset
# If 'wandb' is installed and logged in, training metrics are automatically
# uploaded to the Weights & Biases dashboard.
results = model.train(data="coco8.yaml", epochs=5, project="Ultralytics-WandB-Demo", name="yolo11-experiment")
这种集成可获取系统指标,例如 GPU和模型指标、 提供训练性能的全面视图。如需深入了解,用户可以查看 Ultralytics 集成指南,自定义 记录内容。