fix: release memory after predict (#10688)

* fix: release memory after predict * fix: release memory
2023-08-30 09:22:15 +07:00 · 2023-08-30 09:22:15 +07:00 · 15963b0d24
parent c13a50c7a8
commit 15963b0d24
1 changed files with 1 additions and 0 deletions
--- a/tools/infer/predict_rec.py
+++ b/tools/infer/predict_rec.py
@ -624,6 +624,7 @@ class TextRecognizer(object):
                        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]