参考链接:https://docs.wandb.ai/guides/track/tracking-faq

问题

在加入wandb对训练/推理进行可视化时,由于网络等原因,设置wandb.mode=offline使得代码正常运行。但是W&B网站上却不能同步看到可视化结果,需要手动同步数据。

解决办法

使用同步命令wandb sync wandb/dryrun-folder-name将folder中的数据传入云端,就可以在W&B中看到可视化结果

实例

脚本.py

import wandb
import os

os.environ["WANDB_API_KEY"] = YOUR_KEY_HERE
os.environ["WANDB_MODE"] = "offline"

config = {
  "dataset": "CIFAR10",
  "machine": "offline cluster",
  "model": "CNN",
  "learning_rate": 0.01,
  "batch_size": 128,
}

wandb.init(project="offline-demo")

for i in range(100):
  wandb.log({"accuracy": i})

pycharm中的运行输出

请添加图片描述
其中,wandb sync */wandb/*这行指示了离线的数据保存地址,也是稍后同步时的需要传上去的地址

同步

cd your-wandb-log-folder
wandb sync wandb/offline-run-*-*
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐