wandb.restore()

搜索文档…

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​

以前

wandb.save()

下一个

Import/Export API Guide

最近更新 9mo ago

复制链接

内容

概述

示例