load_dygraph_cn.rst 1.3 KB
Newer Older
1 2 3 4 5
.. _cn_api_fluid_dygraph_load_dygraph:

load_dygraph
-------------------------------

6 7
**注意:该API仅支持【动态图】模式**

8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
.. py:function:: paddle.fluid.dygraph.load_dygraph(model_path)

该接口尝试从磁盘中加载参数或优化器的 ``dict`` 。

该接口会同时加载 ``model_path + ".pdparams"`` 和 ``model_path + ".pdopt"`` 中的内容。

参数:
    - **model_path**  (str) – 保存state_dict的文件前缀。该路径不应该包括后缀 ``.pdparams`` 或 ``.pdopt``。


返回: 两个 ``dict`` ,即从文件中恢复的参数 ``dict`` 和优化器 ``dict``

- para_dict: 从文件中恢复的参数 ``dict``
- opti_dict: 从文件中恢复的优化器 ``dict``

返回类型: tuple(dict, dict)
  
**代码示例**

.. code-block:: python

    import paddle.fluid as fluid

    with fluid.dygraph.guard():
Y
Youwei Song 已提交
32
        emb = fluid.dygraph.Embedding([10, 10])
33 34
        state_dict = emb.state_dict()
        fluid.save_dygraph( state_dict, "paddle_dy")
35 36
        adam = fluid.optimizer.Adam( learning_rate = fluid.layers.noam_decay( 100, 10000) ,
                                     parameter_list = emb.parameters() )
37
        state_dict = adam.state_dict()
38
        fluid.save_dygraph( state_dict, "paddle_dy")
39 40 41 42 43

        para_state_dict, opti_state_dict = fluid.load_dygraph( "paddle_dy")