save_dygraph_cn.rst 1.5 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 34 35
        emb = fluid.dygraph.Embedding(
            size=[10, 32],
            param_attr='emb.w',
            is_sparse=False)
36 37 38
        state_dict = emb.state_dict()
        fluid.save_dygraph(state_dict, "paddle_dy")  # 会保存为 paddle_dy.pdparams

39 40 41
        adam = fluid.optimizer.Adam(
            learning_rate=fluid.layers.noam_decay(100, 10000),
            parameter_list = emb.parameters())
42 43
        state_dict = adam.state_dict()
        fluid.save_dygraph(state_dict, "paddle_dy")  # 会保存为 paddle_dy.pdopt