关于深度学习: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，应用该命令将数据上传到云端