adapt to the develop Paddle and fix export

DrRyanHuang · Nov 7, 2023 · d7785bc · d7785bc
1 parent 3edaaa0
commit d7785bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ppdet/engine/trainer.py b/ppdet/engine/trainer.py
@@ -1018,7 +1018,7 @@ def _get_infer_cfg_and_input_spec(self,
 
         if prune_input:
             static_model = paddle.jit.to_static(
-                self.model, input_spec=input_spec)
+                self.model, input_spec=input_spec, full_graph=True)
             # NOTE: dy2st do not pruned program, but jit.save will prune program
             # input spec, prune input spec here and save with pruned input spec
             pruned_input_spec = _prune_input_spec(