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

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

6

7 8
.. py:function:: paddle.fluid.dygraph.save_dygraph(state_dict, model_path)

S
swtkiwi 已提交
9 10 11 12
:api_attr: 命令式编程模式(动态图)



13 14 15 16 17 18 19
该接口将传入的参数或优化器的 ``dict`` 保存到磁盘上。

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

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

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

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

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

.. code-block:: python

    import paddle.fluid as fluid

    with fluid.dygraph.guard():
35 36
        emb = fluid.dygraph.Embedding([10, 10])

37
        state_dict = emb.state_dict()
38 39 40 41
        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() )
42 43

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