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

Update strategy_config.py

上级 90797a22
...@@ -68,7 +68,7 @@ class QuantAware(BaseStrategy): ...@@ -68,7 +68,7 @@ class QuantAware(BaseStrategy):
window_size=10000, window_size=10000,
moving_rate=0.9, moving_rate=0.9,
for_tensorrt=False, for_tensorrt=False,
onnx_format=False, onnx_format=True,
is_full_quantize=False): is_full_quantize=False):
""" """
Quantization Config. Quantization Config.
...@@ -201,7 +201,7 @@ class QuantPost(BaseStrategy): ...@@ -201,7 +201,7 @@ class QuantPost(BaseStrategy):
activation_quantize_type='range_abs_max', activation_quantize_type='range_abs_max',
simulate_activation_quant=False, simulate_activation_quant=False,
skip_tensor_list=None, skip_tensor_list=None,
onnx_format=False, onnx_format=True,
quantize_op_types=[ quantize_op_types=[
"conv2d", "depthwise_conv2d", "mul", "matmul", "matmul_v2" "conv2d", "depthwise_conv2d", "mul", "matmul", "matmul_v2"
], ],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册