奔溃时释放显存

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