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

代码示例

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")

PaddlePaddle

paddle.static / load_program_state

load_program_state¶

PaddlePaddle

paddle

paddle.amp

paddle.callbacks

paddle.compat

paddle.distributed

paddle.distribution

paddle.fluid

paddle.io

paddle.jit

paddle.metric

paddle.nn

paddle.onnx

paddle.optimizer

paddle.regularizer

paddle.static

paddle.sysconfig

paddle.text

paddle.utils

paddle.vision

paddle.static / load_program_state

load_program_state¶