一、什么是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、生成报告
图片可以通过右上角的生成报告生成