Echo/fcb_photo_review
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

奔溃时释放显存

Browse Source
This commit is contained in:
liuyebo
2024-06-19 16:03:25 +08:00
parent 2e03863fc8
commit 585099521b
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
main.py
View File

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