load_vars

paddle.fluid.io. load_vars ( executor, dirname, main_program=None, vars=None, predicate=None, filename=None )

该接口从文件中加载 Program 的变量。

通过 vars 指定需要加载的变量，或者通过 predicate 筛选需要加载的变量，vars 和 predicate 不能同时为None。

参数

  • executor (Executor) – 运行的执行器，执行器的介绍请参考 模型保存与加载 。

  • dirname (str) – 加载变量所在的目录路径。

  • main_program (Program，可选) – 需要加载变量的 Program ， Program 的介绍请参考 基础概念。如果 main_program 为None，则使用默认的主程序。默认值为None。

  • vars (list[Variable]，可选) – 通过该列表指定需要加载的变量。默认值为None。

  • predicate (function，可选) – 通过该函数筛选 \(predicate(variable)== True\) 的变量进行加载。如果通过 vars 指定了需要加载的变量，则该参数无效。默认值为None。

  • filename (str，可选) – 加载所有变量的文件。如果所有待加载变量是保存在一个文件中，则设置 filename 为该文件名；如果所有待加载变量是按照变量名称单独保存成文件，则设置 filename 为None。默认值为None。

返回

无

抛出异常

  • TypeError - 如果main_program不是Program的实例，也不是None。

代码示例

COPY-FROM: paddle.fluid.io.load_vars