Skip to content

Commit 2d6871d

Browse files
authored
Update quant_layers.py (#35392)
1 parent 7fcb9e3 commit 2d6871d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

python/paddle/nn/quant/quant_layers.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ def __init__(self,
148148
name) if name else 'quant_dequant.scale'
149149
scale_attr = ParamAttr(
150150
name=unique_name.generate(scale_prefix),
151-
initializer=Constant(0.),
151+
initializer=Constant(0.001),
152152
trainable=False)
153153
self._scale = self.create_parameter(
154154
shape=[1], attr=scale_attr, dtype=dtype)
@@ -158,7 +158,7 @@ def __init__(self,
158158
name) if name else 'quant_dequant.state'
159159
state_attr = ParamAttr(
160160
name=unique_name.generate(state_prefix),
161-
initializer=Constant(0),
161+
initializer=Constant(1),
162162
trainable=False)
163163
self._state = self.create_parameter(
164164
shape=[1], attr=state_attr, dtype=dtype)
@@ -168,7 +168,7 @@ def __init__(self,
168168
name) if name else 'quant_dequant.accum'
169169
accum_attr = ParamAttr(
170170
name=unique_name.generate(accum_prefix),
171-
initializer=Constant(0),
171+
initializer=Constant(1),
172172
trainable=False)
173173
self._accum = self.create_parameter(
174174
shape=[1], attr=accum_attr, dtype=dtype)

0 commit comments

Comments
 (0)