load_cn.rst 2.1 KB
Newer Older
Z
zhongpu 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51
.. _cn_api_fluid_load:

load
-------------------------------

.. py:function:: paddle.fluid.load(program, model_path, executor=None, var_list=None)

该接口从Program中过滤出参数和优化器信息,然后从文件中获取相应的值。

如果Program和加载的文件之间参数的维度或数据类型不匹配,将引发异常。

该函数还可以加载用[save_params,save_persistables,save_vars]接口保存的模型文件。
当[save_params,save_persistables,save_vars]保存的模型格式为单个大文件时,var_list不能为None。

参数:
 - **program**  ( :ref:`cn_api_fluid_Program` ) – 要加载的Program。
 - **model_path**  (str) – 保存Program的目录名称+文件前缀。格式为 ``目录名称/文件前缀`` 。
 - **executor** (Executor, 可选) - 当startup program没有运行时,用于初始化参数的Executor。默认值:None。
 - **var_list** (list, 可选) - 指定加载的变量列表,该参数只在加载旧接口[save_params,save_persistables,save_vars]保存的模型文件时使用。当加载的是多个小文件时,变量列表可以是所有加载文件中变量的子集;当加载的单个大文件时,变量列表必须和加载文件中的变量保持一致。

返回: 无

**代码示例**

.. code-block:: python

    # example1
    import paddle.fluid as fluid

    x = fluid.data( name="x", shape=[10, 10], dtype='float32')
    y = fluid.layers.fc(x, 10)
    z = fluid.layers.fc(y, 10)
    place = fluid.CPUPlace()
    exe = fluid.Executor(place)
    exe.run(fluid.default_startup_program())
    fluid.save(fluid.default_main_program(), "./test_path")
    fluid.load(fluid.default_main_program(), "./test_path")

    # example2
    # 注意example1和example2应该分开执行,避免干扰。
    import paddle.fluid as fluid

    x = fluid.data( name="x", shape=[10, 10], dtype='float32')
    y = fluid.layers.fc(x, 10)
    z = fluid.layers.fc(y, 10)
    place = fluid.CPUPlace()
    exe = fluid.Executor(place)
    exe.run(fluid.default_startup_program())
    fluid.save(fluid.default_main_program(), "./test_path")
    fluid.load(fluid.default_main_program(), "./test_path", exe)