From 23760e62d81fec9c7355234a2f39c4f9f5477f64 Mon Sep 17 00:00:00 2001 From: tangwei Date: Fri, 12 Jun 2020 10:48:35 +0800 Subject: [PATCH] fix optimizer SGD --- core/model.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/core/model.py b/core/model.py index bb2040e4..e96d01fe 100755 --- a/core/model.py +++ b/core/model.py @@ -138,9 +138,7 @@ class ModelBase(object): os.environ["FLAGS_communicator_is_sgd_optimizer"] = '0' if name == "SGD": - reg = envs.get_global_env("hyper_parameters.reg", 0.0001) - optimizer_i = fluid.optimizer.SGD( - lr, regularization=fluid.regularizer.L2DecayRegularizer(reg)) + optimizer_i = fluid.optimizer.SGD(lr) elif name == "ADAM": optimizer_i = fluid.optimizer.Adam(lr, lazy_mode=True) elif name == "ADAGRAD": -- GitLab