quickstart:https://wandb.ai/quickstart/p...
$pip install wandb$wandb login "API"import wandbimport 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})# Optionalwandb.watch(model)
训练完结后会在本地生成一个wandb的文件,终端会显示wandb sync wandb/dryrun-folder-name,应用该命令将数据上传到云端