Documentation
搜索文档…
中文 (简体)
wandb.save()
把当前运行的关联文件保存至云端。
有两种方式可以保存与运行关联的文件。
- 1.用
wandb.save(filename)； - 2.将一个文件放到 wandb运行目录下，运行结束后就会被上传。
你如果想在写入文件的同时就同步文件，在
wandb.save中指定一个文件名或通配符。wandb.save示例
1
# Save a model file from the current directory
2
wandb.save('model.h5')
3
4
# Save all files that currently exist containing the substring "ckpt"
5
wandb.save('../logs/*ckpt*')
6
7
# Save any files starting with "checkpoint" as they're written to
8
wandb.save(os.path.join(wandb.run.dir, "checkpoint*"))
Copied!
默认情况下，W&B的本地运行目录位于你的脚本的相对路径./wandb 内，路径名格式如run-20171023_105053-3o4933r0，其中20171023_105053是时间戳，3o4933r0是运行id。你可以将环境变量WANDB_DIR或wandb.init的关键字参数dir，设置为绝对路径，文件就会被写入该路径。
把文件保存至wandb运行项路径示例
文件“model.h5”被保存在wandb.run.dir，训练结束后将被上传。
1
import wandb
2
wandb.init()
3
4
model.fit(X_train, y_train, validation_data=(X_test, y_test),
5
callbacks=[wandb.keras.WandbCallback()])
6
model.save(os.path.join(wandb.run.dir, "model.h5"))
Copied!
这是一个公共示例页面。你可以看到，在“文件”选项卡有个model-best.h5。这是Keras集成默认自动保存的，但你也可以手动保存一个检查点，我们会将其与你的运行关联起来为你保存。
常见问题
忽略某些文件
你可以编辑文件
wandb/settings并设置ignore_globs为一个逗号分割的通配符列表。你也可以设置环境变量WANDB_IGNORE_GLOBS。一个常见的使用情景就是防止我们自动创建的git补丁被上传，WANDB_IGNORE_GLOBS=*.patch 运行结束前同步文件
如果你的运行时间较长，你可能希望看到模型检查点等文件在运行结束前上传到云端。而默认情况下，直到运行结束我们才上传大部分文件。那么，你可以在你的脚本中添加
wandb.save('*.pth')或wandb.save('latest.pth')，以便当这些文件被写入或更新就会被上传。 更改文件保存路径
如果你默认把文件保存至AWS S3或谷歌云存储，你可能会收到这样的错误：
events.out.tfevents.1581193870.gpt-tpu-finetune-8jzqk-2033426287是一个云存储网址，无法将文件保存至wandb要更改TensorBoard事件文件或你希望我们同步的其他文件的记录（log）目录请将你的文件保存到wandb.run.dir以便它们会被同步到我们的云端。
获取运行项名称
如果你要在你的脚本中使用运行名称，你可以用
wandb.run.name，得到运行名称——例如“blissful-waterfall-2”。你要先调用run对象的save，然后才能获得显示名称1
run = wandb.init(...)
2
run.save()
3
print(run.name)
Copied!
把保存的全部文件推送至wandb
在你的脚本开头wandb.init后面调用一次
wandb.save("*.pt")，那么，与该模式匹配的所有文件，被写入wandb.run.dir后就会立即保存。 删除已同步到云存储的本地文件
你可以通过运行命令 wandb gc 来删除已经同步到云存储的本地文件。关于该命令的用法的更多信息，请运行`wandb gc —help
最近更新 9mo ago