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

56次阅读

共计 462 个字符，预计需要花费 2 分钟才能阅读完成。

$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，应用该命令将数据上传到云端

正文完