Wandb 使用

wandb 是帮助记录训练的模型很好用的工具。这篇文章记录了wandb的基础使用过程。

Install & Setup

pip install wandb
wandb login

Init

# init
run = wandb.init(
    # Set the project where this run will be logged
    project="my-awesome-project",
    # Track hyperparameters and run metadata
    config={
        "learning_rate": lr,
        "epochs": epochs,
    })
# take logs use dict
wandb.log({"accuracy": acc, "loss": loss})

Tags

训练时加入 tags 会很方便之后展示的时候使用 filter，更有利于可视化