Skip to content

Commit 7f96abc

Browse files
authored
Merge pull request #6849 from guoshengCS/fix-fluid-lrAttr
Fix lr setting of param_attr in Fluid
2 parents 298dc89 + adfe690 commit 7f96abc

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

python/paddle/v2/fluid/param_attr.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,9 @@ def to_attr(arg):
5858
def to_kwargs(self, with_initializer=False):
5959
kwargs = {
6060
'name': self.name,
61-
'learning_rate': self.learning_rate,
61+
'optimize_attr': {
62+
'learning_rate': self.learning_rate
63+
},
6264
'regularizer': self.regularizer,
6365
'trainable': self.trainable,
6466
'clip_attr': self.clip

0 commit comments

Comments
 (0)