简体中文

English

简体中文

English

Appearance

Weights & Biases ​

Weights & Biases (Wandb) 是一个用于机器学习和深度学习项目的实验跟踪、模型优化和协作平台。W&B 提供了强大的工具来记录和可视化实验结果，帮助数据科学家和研究人员更好地管理和分享他们的工作。

wandb

你可以用两种方式将Wandb上的项目同步到SwanLab：

  1. 同步跟踪：如果你现在的项目使用了wandb进行实验跟踪，你可以使用swanlab.sync_wandb()命令，在运行训练脚本时同步记录指标到SwanLab。
  2. 转换已存在的项目：如果你想要将wandb上的项目复制到SwanLab，你可以使用swanlab convert，将Wandb上已存在的项目转换成SwanLab项目。

INFO

在当前版本暂仅支持转换标量图表。

1. 同步跟踪 ​

1.1 添加sync_wandb命令 ​

在你的代码执行wandb.init()之前的任何位置，添加一行swanlab.sync()命令，即可在训练时同步wandb的指标到SwanLab。

python
import swanlab

swanlab.sync_wandb()

...

wandb.init()

在上述这种代码写法中，wandb.init()的同时会初始化swanlab，项目名、实验名和配置和wandb.init()中的project、name、config一致，因此你不需要再手动初始化swanlab。

1.2 另一种写法 ​

另一种用法是先手动初始化swanlab，再运行wandb的代码。

python
import swanlab

swanlab.init(...)
swanlab.sync_wandb()

...

wandb.init()

在这种写法中，项目名、实验名、配置和swanlab.init()中的project、experiment_name、config一致，而后续wandb.init()中的project、name会被忽略，config会更新进swanlab.config中。

1.3 测试代码 ​

python
import wandb
import random
import swanlab

swanlab.sync_wandb()
# swanlab.init(project="sync_wandb")

wandb.init(
  project="test",
  config={"a": 1, "b": 2},
  name="test",
  )

epochs = 10
offset = random.random() / 5
for epoch in range(2, epochs):
  acc = 1 - 2 ** -epoch - random.random() / epoch - offset
  loss = 2 ** -epoch + random.random() / epoch + offset

  wandb.log({"acc": acc, "loss": loss})

alt text

2. 转换已存在的项目 ​

2.1 找到你在wandb.ai上的projecy、entity和runid ​

projecy、entity和runid是转换所需要的（runid是可选的）。
project和entity的位置： alt text

runid的位置：

alt text

2.2 方式一：命令行转换 ​

首先，需要确保当前环境下，你已登录了wandb，并有权限访问目标项目。

转换命令行：

bash
swanlab convert -t wandb --wb-project [WANDB_PROJECT_NAME] --wb-entity [WANDB_ENTITY]

支持的参数如下：

  • -t: 转换类型，可选wandb与tensorboard。
  • --wb-project：待转换的wandb项目名。
  • --wb-entity：wandb项目所在的空间名。
  • --wb-runid: wandb Run（项目下的某一个实验）的id。

如果不填写--wb-runid，则会将指定项目下的全部Run进行转换；如果填写，则只转换指定的Run。

2.3 方式二：代码内转换 ​

python
from swanlab.converter import WandbConverter

wb_converter = WandbConverter()
# wb_runid可选
wb_converter.run(wb_project="WANDB_PROJECT_NAME", wb_entity="WANDB_USERNAME")

效果与命令行转换一致。