From 15963b0d242867a4cc4d76445626dc8965509b2f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Chung=20Ch=C4=83m=20Ch=E1=BB=89?=
 <45456536+DINHDUCCHUNG@users.noreply.github.com>
Date: Wed, 30 Aug 2023 09:22:15 +0700
Subject: [PATCH] fix: release memory after predict (#10688)

* fix: release memory after predict

* fix: release memory
---
 tools/infer/predict_rec.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tools/infer/predict_rec.py b/tools/infer/predict_rec.py
index 991612ab..5b1fa143 100755
--- 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]
-- 
GitLab