Skip to content

Commit 0edeb83

Browse files
author
Yibing Liu
authored
Fix arg do_model_average in param_attr (#19448)
test=release/1.5
1 parent dd3c242 commit 0edeb83

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

paddle/fluid/API.spec

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -892,7 +892,7 @@ paddle.fluid.LoDTensorArray.append append(self: paddle.fluid.core_avx.LoDTensorA
892892
paddle.fluid.CPUPlace.__init__ __init__(self: paddle.fluid.core_avx.CPUPlace) -> None
893893
paddle.fluid.CUDAPlace.__init__ __init__(self: paddle.fluid.core_avx.CUDAPlace, arg0: int) -> None
894894
paddle.fluid.CUDAPinnedPlace.__init__ __init__(self: paddle.fluid.core_avx.CUDAPinnedPlace) -> None
895-
paddle.fluid.ParamAttr.__init__ (ArgSpec(args=['self', 'name', 'initializer', 'learning_rate', 'regularizer', 'trainable', 'gradient_clip', 'do_model_average'], varargs=None, keywords=None, defaults=(None, None, 1.0, None, True, None, False)), ('document', '6adf97f83acf6453d4a6a4b1070f3754'))
895+
paddle.fluid.ParamAttr.__init__ (ArgSpec(args=['self', 'name', 'initializer', 'learning_rate', 'regularizer', 'trainable', 'gradient_clip', 'do_model_average'], varargs=None, keywords=None, defaults=(None, None, 1.0, None, True, None, True)), ('document', '6adf97f83acf6453d4a6a4b1070f3754'))
896896
paddle.fluid.WeightNormParamAttr.__init__ (ArgSpec(args=['self', 'dim', 'name', 'initializer', 'learning_rate', 'regularizer', 'trainable', 'gradient_clip', 'do_model_average'], varargs=None, keywords=None, defaults=(None, None, None, 1.0, None, True, None, False)), ('document', '6adf97f83acf6453d4a6a4b1070f3754'))
897897
paddle.fluid.DataFeeder.__init__ (ArgSpec(args=['self', 'feed_list', 'place', 'program'], varargs=None, keywords=None, defaults=(None,)), ('document', '6adf97f83acf6453d4a6a4b1070f3754'))
898898
paddle.fluid.DataFeeder.decorate_reader (ArgSpec(args=['self', 'reader', 'multi_devices', 'num_places', 'drop_last'], varargs=None, keywords=None, defaults=(None, True)), ('document', 'be47d7e07824b4281da77472846955ac'))

python/paddle/fluid/param_attr.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,8 @@ class ParamAttr(object):
4242
trainable(bool): Whether this parameter is trainable. Default True.
4343
gradient_clip(BaseGradientClipAttr): The method to clip this parameter's
4444
gradient. Default None.
45-
do_model_average(bool): Whether this parameter should do model average.
46-
Default False.
45+
do_model_average(bool): Whether this parameter should do model average
46+
when model average is enabled. Default True.
4747
4848
Examples:
4949
.. code-block:: python
@@ -65,14 +65,14 @@ def __init__(self,
6565
regularizer=None,
6666
trainable=True,
6767
gradient_clip=None,
68-
do_model_average=False):
68+
do_model_average=True):
6969
self.name = name
7070
self.initializer = initializer
7171
self.learning_rate = learning_rate
7272
self.regularizer = regularizer
7373
self.trainable = trainable
7474
self.gradient_clip = gradient_clip
75-
self.model_average = do_model_average
75+
self.do_model_average = do_model_average
7676

7777
def _set_default_initializer(self, initializer):
7878
"""
@@ -170,7 +170,7 @@ def _to_kwargs(self, with_initializer=False):
170170
'regularizer': self.regularizer,
171171
'trainable': self.trainable,
172172
'gradient_clip_attr': self.gradient_clip,
173-
'model_average': self.model_average
173+
'do_model_average': self.do_model_average
174174
}
175175
if with_initializer:
176176
kwargs['initializer'] = self.initializer

0 commit comments

Comments
 (0)