diff --git a/paddleslim/auto_compression/compressor.py b/paddleslim/auto_compression/compressor.py
index 79e7d91c52d9cb5e8cd88fa9d2f8f200a045c454..de54ad1ce7b3dc10c57e86a28c2a7780b177adbc 100644
--- a/paddleslim/auto_compression/compressor.py
+++ b/paddleslim/auto_compression/compressor.py
@@ -116,7 +116,7 @@ class AutoCompression:
                  as eval_dataloader, and the metric of eval_dataloader for reference only. Dafault: None.
             deploy_hardware(str, optional): The hardware you want to deploy. Default: 'gpu'.
         """
-        self.model_dir = model_dir
+        self.model_dir = model_dir.rstrip('/')
 
         if model_filename == 'None':
             model_filename = None
@@ -144,7 +144,7 @@ class AutoCompression:
         self.train_config = extract_train_config(config)
 
         # prepare dataloader
-        self.feed_vars = get_feed_vars(model_dir, model_filename,
+        self.feed_vars = get_feed_vars(self.model_dir, model_filename,
                                        params_filename)
         self.train_dataloader = wrap_dataloader(train_dataloader,
                                                 self.feed_vars)
@@ -158,7 +158,7 @@ class AutoCompression:
 
         paddle.enable_static()
         self._exe, self._places = self._prepare_envs()
-        self.model_type = self._get_model_type(self._exe, model_dir,
+        self.model_type = self._get_model_type(self._exe, self.model_dir,
                                                model_filename, params_filename)
 
         if self.train_config is not None and self.train_config.use_fleet:
@@ -171,7 +171,7 @@ class AutoCompression:
 
             infer_shape_model = self.create_tmp_dir(
                 self.final_dir, prefix="infer_shape_model_")
-            self._infer_shape(model_dir, self.model_filename,
+            self._infer_shape(self.model_dir, self.model_filename,
                               self.params_filename, input_shapes,
                               infer_shape_model)
             self.model_dir = infer_shape_model