From 585099521baa34a9e138aa9515fd5c2f4fc9ca95 Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Wed, 19 Jun 2024 16:03:25 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A5=94=E6=BA=83=E6=97=B6=E9=87=8A=E6=94=BE?=
 =?UTF-8?q?=E6=98=BE=E5=AD=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/main.py b/main.py
index d764e28..e90a61a 100644
--- a/main.py
+++ b/main.py
@@ -2,6 +2,8 @@ import logging.config
 import os
 import traceback
 
+import paddle
+
 from auto_email.error_email import send_an_error_email
 from config.log import LOGGING_CONFIG
 from config.photo_review import RETRY_TIME, SEND_ERROR_EMAIL
@@ -25,4 +27,6 @@ if __name__ == '__main__':
             if SEND_ERROR_EMAIL:
                 send_an_error_email(program_name='照片审核关键信息抽取脚本', error_name=repr(e),
                                     error_detail=traceback.format_exc())
+            # 释放显存
+            paddle.device.cuda.empty_cache()
             continue