fix: release memory after predict (#10688)

* fix: release memory after predict * fix: release memory

fix: release memory after predict (#10688)
* fix: release memory after predict * fix: release memory
15963b0d · Chung Chăm Chỉ · GitHub · c13a50c7 · 15963b0d
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

tools/infer/predict_rec.py tools/infer/predict_rec.py +1 -0

未找到文件。
--- 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]