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

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

$pip install wandb $wandb login "API"import wandb import osos.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的文件,终端会显示wandb sync wandb/dryrun-folder-name,使用该命令将数据上传到云端

    推荐阅读