深度学习工具——【wandb】offline运行保存wandb的数据文件,再手动同步到W&B网站
在加入wandb对训练/推理进行可视化时,由于网络等原因,设置wandb.mode=offline使得代码正常运行。将folder中的数据传入云端,就可以在W&B中看到可视化结果。这行指示了离线的数据保存地址,也是稍后同步时的需要传上去的地址。上却不能同步看到可视化结果,需要手动同步数据。
文章共581字 · 阅读需要大约2分钟
一键AI生成摘要,助你高效阅读
问答
·
问题
在加入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-*-*
更多推荐
已为社区贡献3条内容
所有评论(0)