未验证 提交 2d6871d3 编写于 作者: X XGZhang 提交者: GitHub

Update quant_layers.py (#35392)

上级 7fcb9e37
...@@ -148,7 +148,7 @@ class FakeQuantMovingAverageAbsMax(layers.Layer): ...@@ -148,7 +148,7 @@ class FakeQuantMovingAverageAbsMax(layers.Layer):
name) if name else 'quant_dequant.scale' name) if name else 'quant_dequant.scale'
scale_attr = ParamAttr( scale_attr = ParamAttr(
name=unique_name.generate(scale_prefix), name=unique_name.generate(scale_prefix),
initializer=Constant(0.), initializer=Constant(0.001),
trainable=False) trainable=False)
self._scale = self.create_parameter( self._scale = self.create_parameter(
shape=[1], attr=scale_attr, dtype=dtype) shape=[1], attr=scale_attr, dtype=dtype)
...@@ -158,7 +158,7 @@ class FakeQuantMovingAverageAbsMax(layers.Layer): ...@@ -158,7 +158,7 @@ class FakeQuantMovingAverageAbsMax(layers.Layer):
name) if name else 'quant_dequant.state' name) if name else 'quant_dequant.state'
state_attr = ParamAttr( state_attr = ParamAttr(
name=unique_name.generate(state_prefix), name=unique_name.generate(state_prefix),
initializer=Constant(0), initializer=Constant(1),
trainable=False) trainable=False)
self._state = self.create_parameter( self._state = self.create_parameter(
shape=[1], attr=state_attr, dtype=dtype) shape=[1], attr=state_attr, dtype=dtype)
...@@ -168,7 +168,7 @@ class FakeQuantMovingAverageAbsMax(layers.Layer): ...@@ -168,7 +168,7 @@ class FakeQuantMovingAverageAbsMax(layers.Layer):
name) if name else 'quant_dequant.accum' name) if name else 'quant_dequant.accum'
accum_attr = ParamAttr( accum_attr = ParamAttr(
name=unique_name.generate(accum_prefix), name=unique_name.generate(accum_prefix),
initializer=Constant(0), initializer=Constant(1),
trainable=False) trainable=False)
self._accum = self.create_parameter( self._accum = self.create_parameter(
shape=[1], attr=accum_attr, dtype=dtype) shape=[1], attr=accum_attr, dtype=dtype)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册