Weights & Biases 深度学习可视化工具

pip install wandb

import wandb

# 使用自己的 W&B 账号登录
wandb.login()

# 开启一个新的 run 来追踪
wandb.init(
    # 指定项目名称
    project="basic-intro",
    # 跟踪超参数并运行元数据
    config={
        "learning_rate": 0.02,
        "architecture": "CNN",
        "dataset": "CIFAR-100",
    }
)

# 在需要显示的地方使用如下方式
wandb.log({
    "acc": acc,
    "loss": loss,
})

# 结束
wandb.finish()