diff --git a/ppcls/engine/engine.py b/ppcls/engine/engine.py
index 7568c16c1..83a24a601 100644
--- a/ppcls/engine/engine.py
+++ b/ppcls/engine/engine.py
@@ -373,7 +373,7 @@ class Engine(object):
                                  "inference")
         if self.quanter:
             self.quanter.save_quantized_model(
-                model,
+                model.base_model,
                 save_path,
                 input_spec=[
                     paddle.static.InputSpec(