fix export quant_model

2025-06-03 21:55:06 +08:00 · 2021-12-03 07:39:31 +00:00 · 2021-12-03 07:39:31 +00:00 · 998f335b40
commit 998f335b40
parent 99b3c96252
1 changed files with 1 additions and 1 deletions
--- 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(