fix: release memory after predict (#10688)

* fix: release memory after predict * fix: release memory
PaddlePaddle · Aug 30, 2023 · 15963b0 · 15963b0
1 parent c13a50c
commit 15963b0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tools/infer/predict_rec.py b/tools/infer/predict_rec.py
@@ -624,6 +624,7 @@ def __call__(self, img_list):
                         preds = outputs
                     else:
                         preds = outputs[0]
+                    self.predictor.try_shrink_memory()
             rec_result = self.postprocess_op(preds)
             for rno in range(len(rec_result)):
                 rec_res[indices[beg_img_no + rno]] = rec_result[rno]