load_persistables

查看属性与别名

API属性：声明式编程(静态图)专用API

paddle.fluid.io.load_persistables ( executor, dirname, main_program=None, filename=None ) [源代码]

该接口从给定的 main_program 中取出所有 persistable==True 的变量（即持久性变量，详见 模型保存与加载 ），并根据目录 dirname 或 filename 提供的参数文件对这些持久性变量进行赋值。

使用 dirname 指定持久性变量的存储路径。若持久性变量以分离文件的形式保存在 dirname 指定的目录下，则设置 filename 值为None；若所有持久性变量保存在一个单独的二进制文件中，则使用 filename 来指明这个二进制文件。

参数

  • executor (Executor) – 加载持久性变量的 executor （详见 执行引擎 ） 。
  • dirname (str) – 持久性变量的存储路径。
  • main_program (Program，可选) – 筛选模型中持久性变量所依据的 Program （详见 基础概念 ）。若为None, 则使用全局默认的 default_main_program 。默认值为None。
  • filename (str，可选) – 若模型中的持久性变量是以若干文件形式存储在 dirname 指定的目录下，则设置 filename 值为None。反之，需要通过 filename 来指明单一模型持久性变量存储文件的名称。 默认值为None。

返回： 无

代码示例

  1. import paddle.fluid as fluid
  3. exe = fluid.Executor(fluid.CPUPlace())
  4. param_path = "./my_paddle_model"
  5. prog = fluid.default_main_program()
  6. fluid.io.load_persistables(executor=exe, dirname=param_path,
  7.                            main_program=None)