Skip to content

Commit 4fdd114

Browse files
committed
a little optimize of optimizer
1 parent fecc9a3 commit 4fdd114

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

python/paddle/fluid/optimizer.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,10 @@ def _create_param_lr(self, param_and_grad):
9292
# create learning rate variable for every parameter
9393
param = param_and_grad[0]
9494
param_lr = param.optimize_attr['learning_rate']
95-
return self.global_learning_rate() * param_lr
95+
if param_lr == 1.0:
96+
return self.global_learning_rate()
97+
else:
98+
return self.global_learning_rate() * param_lr
9699

97100
def _create_accumulators(self, block, parameters):
98101
"""Create all accumulators needed by the parameters

0 commit comments

Comments
 (0)