From 998f335b409b1700b7fa7be089af6d33777c33a9 Mon Sep 17 00:00:00 2001
From: cuicheng01 <cuicheng_smile@163.com>
Date: Fri, 3 Dec 2021 07:39:31 +0000
Subject: [PATCH] fix export quant_model

---
 ppcls/engine/engine.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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(