From 2a44dfb8339ead907b81374a0592c77310457651 Mon Sep 17 00:00:00 2001 From: iLeGend <824040212@qq.com> Date: Fri, 29 Sep 2023 00:03:16 +0800 Subject: [PATCH] fix some --- python/paddle/distributed/sharding/group_sharded.py | 4 ++-- python/paddle/distributed/transpiler/distribute_transpiler.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/python/paddle/distributed/sharding/group_sharded.py b/python/paddle/distributed/sharding/group_sharded.py index 092d99dbb1dcff..b0f5ab0b629cab 100644 --- a/python/paddle/distributed/sharding/group_sharded.py +++ b/python/paddle/distributed/sharding/group_sharded.py @@ -77,7 +77,7 @@ def group_sharded_parallel( Examples: .. code-block:: python - >>> # doctest: +REQUIRES(env:distributed) + >>> # doctest: +REQUIRES(env:DISTRIBUTED) >>> import paddle >>> from paddle.nn import Linear >>> from paddle.distributed import fleet @@ -196,7 +196,7 @@ def save_group_sharded_model(model, output, optimizer=None): Examples: .. code-block:: python - >>> # doctest: +REQUIRES(env:distributed) + >>> # doctest: +REQUIRES(env:DISTRIBUTED) >>> import paddle >>> from paddle.nn import Linear >>> from paddle.distributed import fleet diff --git a/python/paddle/distributed/transpiler/distribute_transpiler.py b/python/paddle/distributed/transpiler/distribute_transpiler.py index fe34951617e212..3d86d6dd9afcef 100644 --- a/python/paddle/distributed/transpiler/distribute_transpiler.py +++ b/python/paddle/distributed/transpiler/distribute_transpiler.py @@ -316,7 +316,7 @@ class DistributeTranspiler: ... pserver_program = t.get_pserver_program(current_endpoint) ... pserver_startup_program = t.get_startup_program(current_endpoint, ... pserver_program) - >>> elif role == "TRAINER": + ... elif role == "TRAINER": ... trainer_program = t.get_trainer_program() >>> # for nccl2 mode