load_program_state¶
从本地加载 Program
的参数和优化器的 Tensor 信息到内存中。
参数¶
model_path (str) - 存储
Program
的参数和优化器的 Tensor 信息的目录名称+文件前缀,格式为目录名称/文件前缀
。var_list (list,可选) - 指定加载的 Tensor 列表,该参数只在加载旧接口[save_params,save_persistables,save_vars]保存的模型文件时使用。当加载的是多个小文件时,Tensor 列表可以是所有加载文件中 Tensor 的子集;当加载的单个大文件时,Tensor 列表必须和加载文件中的 Tensor 保持一致。
返回¶
存储参数和优化器信息的 dict。
代码示例¶
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")
program_state = static.load_program_state("./temp")