load_program_state

paddle.static. load_program_state ( model_path, var_list=None ) [source]
Api_attr

Static Graph

Load program state from local file

Parameters
  • model_path (str) – The file prefix store the program

  • var_list (list, optional) – The Tensor list to load saved with [ save_params, save_persistables, save_vars ]. Default: None. The var_list is only used to get name, will not be modified.

Returns

the dict store Parameter and optimizer information

Return type

state_dict(dict)

Examples

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