未验证 提交 34d97c57 编写于 作者: G Guanghua Yu 提交者: GitHub

fix FakeQuantAbsMax in QAT (#39307)

上级 fdedf909
...@@ -67,7 +67,7 @@ class FakeQuantAbsMax(layers.Layer): ...@@ -67,7 +67,7 @@ class FakeQuantAbsMax(layers.Layer):
if quant_on_weight: if quant_on_weight:
scale_attr = ParamAttr( scale_attr = ParamAttr(
name=self._scale_name, name=self._scale_name,
initializer=Constant(0.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=self._dtype) shape=[1], attr=scale_attr, dtype=self._dtype)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册