a little optimize of optimizer

4fdd114d · qiaolongfei · fecc9a38 · 4fdd114d
显示空白变更内容
内联并排

Showing with 4 addition and 1 deletion

python/paddle/fluid/optimizer.py python/paddle/fluid/optimizer.py +4 -1

未找到文件。
--- a/python/paddle/fluid/optimizer.py
+++ b/python/paddle/fluid/optimizer.py
@@ -92,6 +92,9 @@ class Optimizer(object):
        # create learning rate variable for every parameter
        param = param_and_grad[0]
        param_lr = param.optimize_attr['learning_rate']
+        if param_lr == 1.0:
+            return self.global_learning_rate()
+        else:
            return self.global_learning_rate() * param_lr

    def _create_accumulators(self, block, parameters):