wandb使用很简单,先去官网注册账号,获得一个key,复制
在命令行里
输入pip install wandb下载
wandb login 输入刚才获得的key登陆
在训练代码里面
导入
# Inside my model training code
import wandb
wandb.init(project="my-project")
声明超参数
wandb.config.dropout = 0.2
wandb.config.hidden_layer_size = 128
记录指标
def my_train_loop():
for epoch in range(10):
loss = 0 # change as appropriate :)
wandb.log({'epoch': epoch, 'loss': loss})
保存文件
# by default, this will save to a new subfolder for files associated
# with your run, created in wandb.run.dir (which is ./wandb by default)
wandb.save("mymodel.h5")
# you can pass the full path to the Keras model API
model.save(os.path.join(wandb.run.dir, "mymodel.h5"))
参考
快速上手 https://docs.wandb.ai/v/zh-hans/quickstart
示例: 官方示例代码