set_program_state¶

paddle.static. set_program_state ( program, state_dict ) [source]

Set program parameter from state_dict

An exception will throw if shape or dtype of the parameters is not match.

NOTICE: This function MUST called after run start_up_program

Parameters

• program (Program) – The program to be set

• state_dict (dict) – the dict store Parameter and optimizer information

Returns

None

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

static.set_program_state(prog, program_state)