From 94d5c9b00f926a402bd610f108980a28fa5579fa Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Wed, 26 Jun 2024 12:40:12 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4=E6=98=BE=E5=AD=98=E9=87=8A?=
 =?UTF-8?q?=E6=94=BE=E4=BD=8D=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 photo_review/photo_review.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/photo_review/photo_review.py b/photo_review/photo_review.py
index a87708a..15845f7 100644
--- a/photo_review/photo_review.py
+++ b/photo_review/photo_review.py
@@ -172,6 +172,9 @@ def information_extraction(ie, phrecs):
 
             result = merge_result(result, ie_result)
 
+    # 完成一次任务释放显存
+    paddle.device.cuda.empty_cache()
+    logging.info("释放显存完成")
     return result
 
 
@@ -341,8 +344,6 @@ def main():
                 session.execute(update_flag)
                 session.commit()
                 session.close()
-                # 完成一个案子释放显存
-                paddle.device.cuda.empty_cache()
         else:
             # 没有查询到新案子，等待一段时间后再查
             log = logging.getLogger()