提交 60b33d7c 编写于 作者: R root

Merge branch 'develop' of ssh://gitlab.baidu.com:8022/tianfei01/PaddleSlim into develop

...@@ -26,7 +26,7 @@ WEIGHT_QUANTIZATION_TYPES=['abs_max', 'channel_wise_abs_max'] ...@@ -26,7 +26,7 @@ WEIGHT_QUANTIZATION_TYPES=['abs_max', 'channel_wise_abs_max']
ACTIVATION_QUANTIZATION_TYPES=['abs_max','range_abs_max', 'moving_average_abs_max'] ACTIVATION_QUANTIZATION_TYPES=['abs_max','range_abs_max', 'moving_average_abs_max']
VALID_DTYPES = ['int8'] VALID_DTYPES = ['int8']
quant_config_default = { _quant_config_default = {
# weight quantize type, default is 'abs_max' # weight quantize type, default is 'abs_max'
'weight_quantize_type': 'abs_max', 'weight_quantize_type': 'abs_max',
# activation quantize type, default is 'abs_max' # activation quantize type, default is 'abs_max'
...@@ -59,7 +59,7 @@ def _parse_configs(user_config): ...@@ -59,7 +59,7 @@ def _parse_configs(user_config):
configs(dict): final configs will be used. configs(dict): final configs will be used.
""" """
configs = copy.deepcopy(quant_config_default) configs = copy.deepcopy(_quant_config_default)
configs.update(user_config) configs.update(user_config)
# check configs is valid # check configs is valid
...@@ -209,13 +209,11 @@ def convert(program, scope, place, config, save_int8=False): ...@@ -209,13 +209,11 @@ def convert(program, scope, place, config, save_int8=False):
weight_quantize_type=config['weight_quantize_type']) weight_quantize_type=config['weight_quantize_type'])
freeze_pass.apply(test_graph) freeze_pass.apply(test_graph)
freezed_program = test_graph.to_program() freezed_program = test_graph.to_program()
freezed_program_int8 = None
if save_int8: if save_int8:
convert_int8_pass = ConvertToInt8Pass(scope=fluid.global_scope(), place=place) convert_int8_pass = ConvertToInt8Pass(scope=fluid.global_scope(), place=place)
convert_int8_pass.apply(test_graph) convert_int8_pass.apply(test_graph)
freezed_program_int8 = test_graph.to_program() freezed_program_int8 = test_graph.to_program()
return freezed_program, freezed_program_int8 return freezed_program, freezed_program_int8
else:
return freezed_program
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册