load

paddle.static. load ( program, model_path, executor=None, var_list=None ) [源代码]

从 Program 中过滤出参数和优化器信息,然后从文件中获取相应的值。

如果 Program 和加载的文件之间参数的维度或数据类型不匹配,将引发异常。

该函数还可以加载用[save_params,save_persistables,save_vars]接口保存的模型文件。 当[save_params,save_persistables,save_vars]保存的模型格式为单个大文件时,var_list 不能为 None。

参数

  • program ( Program ) – 要加载的 Program。

  • model_path (str) – 保存 Program 的目录名称+文件前缀。格式为 目录名称/文件前缀

  • executor (Executor,可选) - 当 startup program 没有运行时,用于初始化参数的 Executor。默认值:None。

  • var_list (list,可选) - 指定加载的 Tensor 列表,该参数只在加载旧接口[save_params,save_persistables,save_vars]保存的模型文件时使用。当加载的是多个小文件时,Tensor 列表可以是所有加载文件中 Tensor 的子集;当加载的单个大文件时,Tensor 列表必须和加载文件中的 Tensor 保持一致。

返回

无。

代码示例

>>> import paddle
>>> import paddle.static as static

>>> paddle.enable_static()

>>> x = static.data(name="x", shape=[10, 10], dtype='float32')
>>> y = static.nn.fc(x, 10)
>>> z = static.nn.fc(y, 10)

>>> place = paddle.CPUPlace()
>>> exe = static.Executor(place)
>>> exe.run(static.default_startup_program())
>>> prog = static.default_main_program()

>>> static.save(prog, "./temp")
>>> static.load(prog, "./temp")