Documentation
搜索文档…
wandb.restore()
还原文件,例如模型检查点,还原至本地运行项文件夹,以供脚本使用。

概述

调用wandb.restore(filename)将把文件还原至本地运行项路径。一般情况下,filename指的是之前的实验运行项生成的文件,保存在我们的云端。这一调用将把该文件复制到本地,并返回一个本地文件流,该文件流开放供读取。
wandb.restore可以接受几个可选关键字参数:
  • run_path — s字符串型,指的是之前的运行项,要把文件复制到这一运行项,格式为“$ENTITY_NAME/$PROJECT_NAME/$RUN_ID' or '$PROJECT_NAME/$RUN_ID”(默认值:当前的归属单位、项目名称和运行项id)
  • replace — 布尔型,如果本地有同名文件,指定是否用云端文件覆盖本地的同名文件。(默认值:False)
  • root — 字符串型,指定把云端文件还原至本地的哪个路径。默认值为当前工作路径,如果前面调用过wandb.init,默认值为wandb.run.dir。(默认值:“.”)
常见使用情景:
  • 还原之前的运行项生成的模型架构和权值。
  • 出现故障之后,从上一个检查点开始断点续训。(进入断点续训板块详细了解)

示例

若要查看完整的运行示例,请看这篇报告
1
# restore a model file from a specific run by user "vanpelt" in "my-project"
2
best_model = wandb.restore('model-best.h5', run_path="vanpelt/my-project/a1b2c3d")
3
4
# restore a weights file from a checkpoint
5
# (NOTE: resuming must be configured if run_path is not provided)
6
weights_file = wandb.restore('weights.h5')
7
# use the "name" attribute of the returned object
8
# if your framework expects a filename, e.g. as in Keras
9
my_predefined_model.load_weights(weights_file.name)
Copied!
如果你不指定run_path,就需要为运行项配置断点续训。如果你想在训练之外用编程方式访问文件,就用运行项API
最近更新 9mo ago
复制链接