set_program_state

  • paddle.fluid.io.set_program_state(program, state_dict)[源代码]

利用 state_dict 设置 Program 的参数和优化器信息。

如果参数的 shape 或 dtype 不匹配，则会引发异常。

注意：必须在运行 start_up_program 之后调用此函数。

  • 参数:
    • program (Program) - 需要被设置的 Program 。
    • state_dict (dict) - 存储参数和优化器信息的dict；dict中key的类型为变量的名称，value为np.ndarray类型的数据。

返回：无

代码示例

  1. import paddle.fluid as fluid
  2.  
  3. x = fluid.data( name="x", shape=[10, 10], dtype='float32')
  4. y = fluid.layers.fc( x, 10)
  5. z = fluid.layers.fc( y, 10)
  6.  
  7. place = fluid.CPUPlace()
  8. exe = fluid.Executor(place)
  9. exe.run( fluid.default_startup_program() )
  10. prog = fluid.default_main_program()
  11.  
  12. fluid.save( prog, "./temp")
  13. program_state = fluid.load_program_state( "./temp")
  14. fluid.set_program_state( prog, program_state)