diff --git a/ce_tests/dygraph/quant/src/save_quant_model.py b/ce_tests/dygraph/quant/src/save_quant_model.py index 0ca94cf3f95b8172fce333955228612bc7dd7e68..acf36c93c8fc1cd069219a56a03551b818b7bfeb 100644 --- a/ce_tests/dygraph/quant/src/save_quant_model.py +++ b/ce_tests/dygraph/quant/src/save_quant_model.py @@ -23,7 +23,7 @@ import numpy as np import time import paddle from paddle.fluid.framework import IrGraph -from paddle.quantization import Quant2Int8MkldnnPass +from paddle.static.quantization import Quant2Int8MkldnnPass from paddle.framework import core paddle.enable_static()