基于pytorch的 wandb使用

一、什么是wandb

wandb是一个云数据记录追踪工具
https://wandb.ai

二、 使用步骤

1、登录网站，创建自己的账户（可使用GitHub账号登陆）
2、创建一个项目

3、输入项目名称

4、安装函数包

pip install wandb

5、在训练程序中引入函数包

import wandbl

参考train_wandb.py
6、加入超参数

wandb.config = {
  "learning_rate": 0.001,
  "epochs": 100,
  "batch_size": 128
}#字典形式加入任意你需要的超参数

7、添加需要观察的变量

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

# Optional
wandb.watch(model)

8、运行程序

conda activate your-enviroment#激活你的虚拟环境
wandb login your-key#复制下面的那一串数字
python train_wandbl.py

9、观察结果
运行后会在命令行给出网址，复制到浏览器即可打开


生成的表格横坐标默认是Step，如果想更改为Epoch
wandb.log({“loss”: loss,“epoch”: epoch，})

如此即可。
可以在这里更改线段的名称

10、生成报告

图片可以通过右上角的生成报告生成