Skip to content

Commit 3e8b107

Browse files
authored
Update trainer.py (#8696)
fix #8692
1 parent 4877a75 commit 3e8b107

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ppdet/engine/trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1179,7 +1179,7 @@ def _get_infer_cfg_and_input_spec(self,
11791179
})
11801180
if prune_input:
11811181
static_model = paddle.jit.to_static(
1182-
self.model, input_spec=input_spec)
1182+
self.model, input_spec=input_spec, full_graph=True)
11831183
# NOTE: dy2st do not pruned program, but jit.save will prune program
11841184
# input spec, prune input spec here and save with pruned input spec
11851185
pruned_input_spec = _prune_input_spec(

0 commit comments

Comments
 (0)