wandb深度学习记录训练数据工具初体验

quickstart：https://wandb.ai/quickstart/p...

$pip install wandb
$wandb login "API"

import wandb
import os

os.environ["WANDB_API_KEY"] = "API"
os.environ["WANDB_MODE"] = "offline"  #离线保存数据的方法：因网络问题无法在线保存，可将数据先保存在本地，在同步到云端

wandb.init(project="my-test-project")
wandb.config = {
  "learning_rate": 0.001,
  "epochs": 100,
  "batch_size": 128
}

for epoch in range(epoches): #训练循环中打印
    wandb.log({"loss": loss})

# Optional
wandb.watch(model)

训练结束后会在本地生成一个wandb的文件，终端会显示wandb sync wandb/dryrun-folder-name，使用该命令将数据上传到云端