提交 3df76b0f 编写于 作者: G guofei 提交者: Zeng Jinle

Rewrite the content of save_persistables_cn.rst develop=test (#1369)

* fix save_persistables  develop=test

* fix save_persistables develop=test

* rewrite save_persistables_cn.rst test=develop
上级 cf8fbdd7
...@@ -5,17 +5,17 @@ save_persistables ...@@ -5,17 +5,17 @@ save_persistables
.. py:function:: paddle.fluid.io.save_persistables(executor, dirname, main_program=None, filename=None) .. py:function:: paddle.fluid.io.save_persistables(executor, dirname, main_program=None, filename=None)
函数从给定 ``main_program`` 中取出所有 ``persistable==True`` 的变量,然后将它们保存到目录 ``dirname`` 中或 ``filename`` 指定的文件中。 OP从给定 ``main_program`` 中取出所有持久性变量(详见 :ref:`api_guide_model_save_reader` ),然后将它们保存到目录 ``dirname`` 中或 ``filename`` 指定的文件中。
``dirname`` 用于指定保存长期变量的目录。如果想将变量保存到指定目录的若干文件中,设置 ``filename=None`` ; 如果想将所有变量保存在一个文件中,请使用 ``filename`` 来指定它 ``dirname`` 用于指定保存持久性变量的目录。如果想将持久性变量保存到指定目录的若干文件中,请设置 ``filename=None`` ; 若想将所有持久性变量保存在同一个文件中,请设置 ``filename`` 来指定文件的名称
参数: 参数:
- **executor** (Executor) – 保存变量的 executor - **executor** (Executor) – 用于保存持久性变量的 ``executor`` ,详见 :ref:`api_guide_executor` 。
- **dirname** (str) – 目录路径 - **dirname** (str) – 用于储存持久性变量的文件目录。
- **main_program** (Program|None) – 需要保存变量的 Program。如果为 None,则使用 default_main_Program 。默认值: None - **main_program** (Program,可选) – 需要保存持久性变量的Program( ``Program`` 含义详见 :ref:`api_guide_Program` )。如果为None,则使用default_main_Program 。默认值为None。
- **filename** (str|None) – 保存变量的文件。如果想分开保存变量,设置 filename=None. 默认值: None - **filename** (str,可选) – 保存持久性变量的文件名称。若想分开保存变量,设置 ``filename=None`` 。 默认值为None。
返回: None 返回:
**代码示例** **代码示例**
...@@ -23,13 +23,20 @@ save_persistables ...@@ -23,13 +23,20 @@ save_persistables
import paddle.fluid as fluid import paddle.fluid as fluid
exe = fluid.Executor(fluid.CPUPlace()) dir_path = "./my_paddle_model"
param_path = "./my_paddle_model" file_name = "persistables"
prog = fluid.default_main_program() image = fluid.layers.data(name='img', shape=[1, 28, 28], dtype='float32')
# `prog` 可以是由用户自定义的program label = fluid.layers.data(name='label', shape=[1], dtype='int64')
fluid.io.save_persistables(executor=exe, dirname=param_path, feeder = fluid.DataFeeder(feed_list=[image, label], place=fluid.CPUPlace())
main_program=prog) predict = fluid.layers.fc(input=image, size=10, act='softmax')
loss = fluid.layers.cross_entropy(input=predict, label=label)
avg_loss = fluid.layers.mean(loss)
exe = fluid.Executor(fluid.CPUPlace())
exe.run(fluid.default_startup_program())
fluid.io.save_persistables(executor=exe, dirname=dir_path, filename=file_name)
# 网络中fc层中的持久性变量weight和bia将会保存在路径“./my_paddle_model”下名为"persistables"的文件中。
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册