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

fix qat onnx_format pass (#1440)

上级 71e0acb1
...@@ -43,7 +43,7 @@ try: ...@@ -43,7 +43,7 @@ try:
from paddle.fluid.contrib.slim.quantization import QuantWeightPass from paddle.fluid.contrib.slim.quantization import QuantWeightPass
from paddle.fluid.contrib.slim.quantization import AddQuantDequantPassV2 from paddle.fluid.contrib.slim.quantization import AddQuantDequantPassV2
from paddle.fluid.contrib.slim.quantization import PostTrainingQuantizationProgram from paddle.fluid.contrib.slim.quantization import PostTrainingQuantizationProgram
from paddle.fluid.contrib.slim.quantization import OutScaleForInferencePassV2 from paddle.fluid.contrib.slim.quantization import AddQuantDequantForInferencePass
except: except:
_logger.warning( _logger.warning(
"Some functions fail to import, please update PaddlePaddle version to 2.4+" "Some functions fail to import, please update PaddlePaddle version to 2.4+"
...@@ -686,7 +686,7 @@ def convert(program, ...@@ -686,7 +686,7 @@ def convert(program,
quant_weight_pass = QuantWeightPass(scope, place) quant_weight_pass = QuantWeightPass(scope, place)
quant_weight_pass.apply(test_graph) quant_weight_pass.apply(test_graph)
try: try:
out_scale_infer_pass = OutScaleForInferencePassV2( out_scale_infer_pass = AddQuantDequantForInferencePass(
scope=scope, place=place, quant_bits=config['activation_bits']) scope=scope, place=place, quant_bits=config['activation_bits'])
out_scale_infer_pass.apply(test_graph) out_scale_infer_pass.apply(test_graph)
except: except:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册