fix export quant_model

998f335b · cuicheng01 · 99b3c962 · 998f335b
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppcls/engine/engine.py ppcls/engine/engine.py +1 -1

未找到文件。
--- 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(