load_program_state

paddle.static. load_program_state ( model_path, var_list=None )

从本地加载 Program 的参数和优化器的 Tensor 信息到内存中。

参数

  • model_path (str) - 存储 Program 的参数和优化器的 Tensor 信息的目录名称+文件前缀,格式为 目录名称/文件前缀

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

返回

存储参数和优化器信息的 dict。

代码示例

COPY-FROM: paddle.static.load_program_state