load_program_state

paddle.fluid.io.load_program_state(model_path, var_list=None)[源代码]

该接口从本地加载 Program 的参数和优化器的变量信息到内存中。

参数

  • model_path (str) - 存储 Program 的参数和优化器的变量信息的目录名称+文件前缀，格式为 目录名称/文件前缀 。
  • var_list (list, 可选) - 指定加载的变量列表，该参数只在加载旧接口[save_params，save_persistables，save_vars]保存的模型文件时使用。当加载的是多个小文件时，变量列表可以是所有加载文件中变量的子集；当加载的单个大文件时，变量列表必须和加载文件中的变量保持一致。

返回

存储参数和优化器信息的dict

返回类型

dict

代码示例

  1. import paddle.fluid as fluid
  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)
  7. place = fluid.CPUPlace()
  8. exe = fluid.Executor(place)
  9. exe.run( fluid.default_startup_program() )
  10. prog = fluid.default_main_program()
  12. fluid.save( prog, "./temp")
  13. program_state = fluid.load_program_state( "./temp")