diff --git a/python/paddle/fluid/contrib/slim/quantization/imperative/utils.py b/python/paddle/fluid/contrib/slim/quantization/imperative/utils.py index 004e1c1aa9bc501c4a91be5b821ce505592f6910..491f8a7e25cbcd451bed0f5daef20b225f1c6a1f 100644 --- a/python/paddle/fluid/contrib/slim/quantization/imperative/utils.py +++ b/python/paddle/fluid/contrib/slim/quantization/imperative/utils.py @@ -39,7 +39,7 @@ quant_input_layers_map = { fake_quantize_dequantize_types = [ "fake_quantize_dequantize_abs_max", - "fake_quantize_dequantize_channel_wise_abs_max", + "fake_channel_wise_quantize_dequantize_abs_max", "fake_quantize_dequantize_moving_average_abs_max" ]