问题
在加入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-*-*