From 5a521800976ae301c9da1606ce6631555b99d507 Mon Sep 17 00:00:00 2001 From: Chengmo Date: Wed, 17 Jun 2020 14:24:37 +0800 Subject: [PATCH] Update doc & fix setup (#100) * fix doc * fix setup Co-authored-by: wuzhihua <35824027+fuyinno4@users.noreply.github.com> --- doc/custom_reader.md | 2 +- doc/trainer_develop.md | 10 +++++----- setup.py | 2 +- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/doc/custom_reader.md b/doc/custom_reader.md index 748f82c5..1d95f868 100644 --- a/doc/custom_reader.md +++ b/doc/custom_reader.md @@ -77,7 +77,7 @@ Reader的逻辑需要一个单独的python文件进行描述。我们试写一 ``` 2. 创建一个子类，继承Reader的基类，训练所需Reader命名为`TrainerReader` ```python - class TrainerReader(Reader): + class TrainerReader(ReaderBase): def init(self): pass diff --git a/doc/trainer_develop.md b/doc/trainer_develop.md index 1b9f51b0..4d93149c 100644 --- a/doc/trainer_develop.md +++ b/doc/trainer_develop.md @@ -26,7 +26,7 @@ class InstanceBase(object): ## network -network将在instanc后调用，执行模型的组网。network的官方实现位于[network.py](../core/trainers/framework/network.py),network基类定义如下： +network将在instanc后调用，执行模型的组网。network的官方实现位于[network.py](../core/trainers/framework/network.py)，network基类定义如下： ```python class NetworkBase(object): @@ -37,7 +37,7 @@ class NetworkBase(object): pass ``` -可参照其他模式的实现方式，自定其中的部分步骤。您需要您需要继承`NetworkBase`并命名为`Network`，完成`build_network`的实现，通过上下文信息字典`context`拿到模型所需信息，并在context中保存模型的program与scope信息，例如： +可参照其他模式的实现方式，实现其中的部分步骤。您需要继承`NetworkBase`并命名为`Network`，完成`build_network`的实现，通过上下文信息字典`context`拿到模型所需信息，并在context中保存模型的program与scope信息，例如： ```python context["model"][model_dict["name"]][ @@ -52,7 +52,7 @@ context["model"][model_dict["name"]][ ## startup -startup执行网络参数的初始化，抑或模型的热启动，主要功能是执行`exe.run(fluid.default_startup_program())`。 startup的官方实现在[startup](../core/trainers/framework/startup.py) +startup执行网络参数的初始化，或者模型的热启动，主要功能是执行`exe.run(fluid.default_startup_program())`。 startup的官方实现在[startup](../core/trainers/framework/startup.py) ```python class StartupBase(object): @@ -75,7 +75,7 @@ class StartupBase(object): context["exe"], dirname, main_program=main_program) ``` -自定义startup流程，您需要您需要继承`StartupBase`并命名为`Startup`，实现该类型中startup成员函数。 +自定义startup流程，您需要继承`StartupBase`并命名为`Startup`，实现该类型中startup成员函数。 ## runner @@ -113,7 +113,7 @@ class PSRunner(RunnerBase): context["status"] = "terminal_pass" ``` -自定义runner需要参照官方实现[runner.py](../core/trainers/framework/startup.py)，继承基类`RunnerBase`，命名为`Runner`，并实现`run`成员函数。 +自定义runner需要参照官方实现[runner.py](../core/trainers/framework/runner.py)，继承基类`RunnerBase`，命名为`Runner`，并实现`run`成员函数。 ## terminal diff --git a/setup.py b/setup.py index 60127534..6e4b22ec 100644 --- a/setup.py +++ b/setup.py @@ -22,7 +22,7 @@ from setuptools import setup, find_packages import shutil import tempfile -requires = ["paddlepaddle >= 1.7.2", "PyYAML >= 5.1.1"] +requires = ["paddlepaddle == 1.7.2", "PyYAML >= 5.1.1"] about = {} about["__title__"] = "paddle-rec" -- GitLab