diff --git a/paddleslim/auto_compression/auto_strategy.py b/paddleslim/auto_compression/auto_strategy.py index a1eb0b0f2df7282826d63c58311a571a18d5fdb8..2826601aeccac65adf385b45a7bcc7e1eb842800 100644 --- a/paddleslim/auto_compression/auto_strategy.py +++ b/paddleslim/auto_compression/auto_strategy.py @@ -241,16 +241,6 @@ def prepare_strategy(executor, def get_final_quant_config(ptq_loss, model_type=None): """ transform quantization tester config to real quantization config """ - ### use ptq & hpo when model_type is transformer - if model_type == 'transformer': - quant_config = Quantization(**default_quant_config) - hpo_config = HyperParameterOptimization(**default_hpo_config) - configs = [{ - 'Quantization': quant_config, - 'HyperParameterOptimization': hpo_config - }] - return configs - ### if emd loss less than MAGIC_MIN_EMD_DISTANCE, final compress. if ptq_loss < MAGIC_MIN_EMD_DISTANCE: return None