fix: release memory after predict (PaddlePaddle#10688)

* fix: release memory after predict * fix: release memory
truongnn1106 · 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]