refine auto strategy, test=document_fix (#28211)

4 years ago · 11acbfae06
parent 2db77be423
commit 11acbfae06
1 changed files with 4 additions and 0 deletions
--- a/python/paddle/distributed/fleet/base/distributed_strategy.py
+++ b/python/paddle/distributed/fleet/base/distributed_strategy.py
@ -1073,8 +1073,12 @@ class DistributedStrategy(object):

            import paddle
            import paddle.distributed.fleet as fleet
+            paddle.enable_static()
+
            strategy = fleet.DistributedStrategy()
            strategy.auto = True
+            # if set other strategy at the same time, auto will not apply
+            # strategy.amp = True

            optimizer = paddle.optimizer.SGD(learning_rate=0.01)
            optimizer = fleet.distributed_optimizer(optimizer, strategy)