三行代码可视化和记录炼丹过程（wandb快速教程）

wandb简介

[wandb] is short of ‘Experiment Tracking with Weights and Biases’. The authors are Biewald and Lukas, and the software is available from wandb.com .

​快速开始

1. 访问https://wandb.ai/ 并注册。很方便直接用Google/Github账号。

2. 安装wandb并登录：

pip install wandb

wandb login

三行代码

import wandb

...

wandb.init(project="PRO_NAME", id="ID_NAME")

...

 wandb.log({"loss": loss, "acc5": acc5})

如果多卡训练，在一张卡上初始化即可。现在你就可以在https://wandb.ai/ 看到曲线、命令和一堆运行信息了。

​如果想记录所有的argparses，可以加第四行代码：

wandb.config.update(args)

就可以在信息栏看到这个了，妈妈再也不用担心我记错实验结果对应的设置了！