fix a quantization bug (#33753)

57352bc7 · XGZhang · GitHub · af9dcb2d · 57352bc7
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/contrib/slim/quantization/quantization_pass.py ...ddle/fluid/contrib/slim/quantization/quantization_pass.py +1 -1

未找到文件。
--- a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
+++ b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
@@ -1148,7 +1148,7 @@ class QuantizationFreezePass(object):
                    ], "the dim of scale_v should be 1 or 2"
                    if scale_v.ndim == 2:
                        scale_v = scale_v[0]
-                    if scale_v.size == 1:
+                    if scale_v.size == 1 and self._weight_quantize_type == 'abs_max':
                        scale_v = scale_v[0]
                    else:
                        scale_v = scale_v.tolist()