save_dygraph_cn.rst 1.4 KB
Newer Older
1 2 3 4 5
.. _cn_api_fluid_dygraph_save_dygraph:

save_dygraph
-------------------------------

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

8 9 10 11 12 13 14 15 16
.. py:function:: paddle.fluid.dygraph.save_dygraph(state_dict, model_path)

该接口将传入的参数或优化器的 ``dict`` 保存到磁盘上。

``state_dict`` 是通过 :ref:`cn_api_fluid_dygraph_Layer` 的 ``state_dict()`` 方法得到的。

注: ``model_path`` 不可以是一个目录。

该接口会根据 ``state_dict`` 的内容,自动给 ``model_path`` 添加 ``.pdparams`` 或者 ``.pdopt`` 后缀,
17
生成 ``model_path + ".pdparams"`` 或者 ``model_path + ".pdopt"`` 文件。
18 19 20 21 22 23 24 25 26 27 28 29 30 31

参数:
 - **state_dict**  (dict of Parameters) – 要保存的模型参数的 ``dict`` 。
 - **model_path**  (str) – 保存state_dict的文件前缀。格式为 ``目录名称/文件前缀``。如果文件前缀为空字符串,会引发异常。

返回: 无
  
**代码示例**

.. code-block:: python

    import paddle.fluid as fluid

    with fluid.dygraph.guard():
32 33
        emb = fluid.dygraph.Embedding([10, 10])

34
        state_dict = emb.state_dict()
35 36 37 38
        fluid.save_dygraph( state_dict, "paddle_dy") # 会保存为 paddle_dy.pdparams

        adam = fluid.optimizer.Adam( learning_rate = fluid.layers.noam_decay( 100, 10000),
                                     parameter_list = emb.parameters() )
39 40

        state_dict = adam.state_dict()
41
        fluid.save_dygraph( state_dict, "paddle_dy") # 会保存为 paddle_dy.pdopt