fix dist_ctr getenv, test=develop

7 years ago · 877289c4ca
parent 1e04222890
commit 877289c4ca
1 changed files with 2 additions and 2 deletions
--- a/python/paddle/fluid/tests/unittests/dist_ctr.py
+++ b/python/paddle/fluid/tests/unittests/dist_ctr.py
@ -99,10 +99,10 @@ class TestDistCTR2x2(TestDistRunnerBase):
        inference_program = paddle.fluid.default_main_program().clone()

        regularization = None
-        use_l2_decay = bool(os.getenv(['USE_L2_DECAY'], 0))
+        use_l2_decay = bool(os.getenv('USE_L2_DECAY', 0))
        if use_l2_decay:
            regularization = fluid.regularizer.L2DecayRegularizer(
-                regularization_coeff=1e-3)
+                regularization_coeff=1e-1)

        sgd_optimizer = fluid.optimizer.SGD(learning_rate=0.0001,
                                            regularization=regularization)