From fa949b1b7be17cac809542391be691e6d63d1fdf Mon Sep 17 00:00:00 2001
From: Guanghua Yu <742925032@qq.com>
Date: Fri, 7 Apr 2023 14:00:39 +0800
Subject: [PATCH] fix qat export bug when input_spec is None (#52615)

---
 python/paddle/quantization/imperative/qat.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/paddle/quantization/imperative/qat.py b/python/paddle/quantization/imperative/qat.py
index 9a1109ccbf6..591dac54507 100644
--- a/python/paddle/quantization/imperative/qat.py
+++ b/python/paddle/quantization/imperative/qat.py
@@ -530,7 +530,8 @@ class ImperativeQuantizeOutputs:
             model, paddle.nn.Layer
         ), "The model must be the instance of paddle.nn.Layer."
 
-        paddle.jit.to_static(model, input_spec=input_spec)
+        if input_spec:
+            paddle.jit.to_static(model, input_spec=input_spec)
         paddle.jit.save(layer=model, path=path, input_spec=input_spec, **config)
 
         is_dynamic_mode = False
-- 
GitLab