From c3d32424efd96f7047f55f4100af682da83efba0 Mon Sep 17 00:00:00 2001 From: jingqinghe Date: Fri, 14 Aug 2020 11:02:43 +0800 Subject: [PATCH] add unittest --- python/paddle/fluid/tests/unittests/test_fleet_base.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_fleet_base.py b/python/paddle/fluid/tests/unittests/test_fleet_base.py index 8019841f72e..2292b06303e 100644 --- a/python/paddle/fluid/tests/unittests/test_fleet_base.py +++ b/python/paddle/fluid/tests/unittests/test_fleet_base.py @@ -145,9 +145,8 @@ class TestFleetBase(unittest.TestCase): import paddle.fluid.incubate.fleet.base.role_maker as role_maker role = role_maker.PaddleCloudRoleMaker(is_collective=True) fleet.init(role) - strategy = fleet.DistributedStrategy() optimizer = paddle.optimizer.SGD(learning_rate=0.001) - optimizer = fleet.distributed_optimizer(optimizer, strategy=strategy) + optimizer = fleet.distributed_optimizer(optimizer) def test_minimize(self): import paddle -- GitLab