diff --git a/paddleslim/auto_compression/compressor.py b/paddleslim/auto_compression/compressor.py index 7d21bbe7a13d01f660826864285d14ea235aabf3..5885387f0e325a6fc11aaf68f83ba58a63665503 100644 --- a/paddleslim/auto_compression/compressor.py +++ b/paddleslim/auto_compression/compressor.py @@ -816,8 +816,14 @@ class AutoCompression: for name in test_program_info.feed_target_names ] - model_name = '.'.join(self.model_filename.split( - '.')[:-1]) if self.model_filename is not None else 'model' + model_name = None + if self.model_filename is None: + model_name = "model" + elif self.model_filename.endswith(".pdmodel"): + model_name = self.model_filename.rsplit(".", 1)[0] + else: + model_name = self.model_filename + path_prefix = os.path.join(model_dir, model_name) paddle.static.save_inference_model( path_prefix=path_prefix,